tnt306 commited on
Commit
8a96d81
·
1 Parent(s): 09980bd

Updated to latest trained model

Browse files
Files changed (2) hide show
  1. app.py +1 -1
  2. 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.7033 (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
  """
 
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:fc8ce76b804f59492a8898b568978aaf73a64bcd8ab7e97cbf83626113ffde39
3
- size 60944806
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:fe03574b02ac988557f12d79734fc7057707121ad9b09a1b2fb30870486df91c
3
+ size 60944146