Spaces:
Sleeping
Sleeping
Updated to latest trained model
Browse files- app.py +1 -1
- models/final_model.pt +2 -2
app.py
CHANGED
@@ -245,7 +245,7 @@ with gr.Blocks(theme=gr.themes.Soft(), css=css) as demo:
|
|
245 |
- 5,315 positive, 44,999 negative (11.8%)
|
246 |
- Split: 80% training, 10% validation, 10% testing
|
247 |
- Notable points:
|
248 |
-
- Result: AUPRC 0.
|
249 |
- Variational Regularization, inspired by [Kipf et al., 2016](https://arxiv.org/abs/1611.07308)
|
250 |
- Trained on NVIDIA A100 with PyTorch 2.4.0
|
251 |
"""
|
|
|
245 |
- 5,315 positive, 44,999 negative (11.8%)
|
246 |
- Split: 80% training, 10% validation, 10% testing
|
247 |
- Notable points:
|
248 |
+
- Result: AUPRC 0.7027 (Baseline: 0.118) on Val split
|
249 |
- Variational Regularization, inspired by [Kipf et al., 2016](https://arxiv.org/abs/1611.07308)
|
250 |
- Trained on NVIDIA A100 with PyTorch 2.4.0
|
251 |
"""
|
models/final_model.pt
CHANGED
@@ -1,3 +1,3 @@
|
|
1 |
version https://git-lfs.github.com/spec/v1
|
2 |
-
oid sha256:
|
3 |
-
size
|
|
|
1 |
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:fe03574b02ac988557f12d79734fc7057707121ad9b09a1b2fb30870486df91c
|
3 |
+
size 60944146
|