vgoradia commited on
Commit
44d0475
·
verified ·
1 Parent(s): a244950

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +20 -0
README.md CHANGED
@@ -1,3 +1,23 @@
1
  ---
2
  license: apache-2.0
3
  ---
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
  ---
2
  license: apache-2.0
3
  ---
4
+ ---
5
+ license: apache-2.0
6
+ ---
7
+
8
+ # MolScreen
9
+
10
+ Graph neural network for early-stage molecular toxicity screening.
11
+
12
+ Trained on the Tox21 dataset (7,823 compounds, 12 toxicity endpoints).
13
+ Test mean ROC-AUC: 0.8525.
14
+
15
+ **Live app:** https://mol-screen-32jrti5gpuajmd8wse5u3f.streamlit.app
16
+ **GitHub:** https://github.com/vgoradia/mol-screen
17
+
18
+ ## Usage
19
+
20
+ Load with PyTorch:
21
+
22
+ model = MolScreenGNN()
23
+ model.load_state_dict(torch.load('molscreen_best.pt'))