Skip to content

Commit

Permalink
Update
Browse files Browse the repository at this point in the history
  • Loading branch information
mufeili committed Jan 21, 2021
1 parent b0dd3ab commit 2cc0328
Show file tree
Hide file tree
Showing 3 changed files with 17 additions and 0 deletions.
3 changes: 3 additions & 0 deletions README.md
Original file line number Diff line number Diff line change
Expand Up @@ -28,3 +28,6 @@
### ClinTox

| Rank | Model | Featurization | Test ROC-AUC | Validation ROC-AUC | Contact | References | Date |
| ---- | ------------- | -------------- | ---------------- | ------------------ | --------------------------------- | ---------------------------------------------------------------------------------------- | -------------- |
| 1 | GCN | GraphConv | 0.9065 +- 0.0179 | 0.9880 +- 0.0073 | [Mufei Li]([email protected]) | [Paper](https://arxiv.org/abs/1609.02907), [Code](./examples) | Jan 22nd, 2021 |
| 2 | Random Forest | 1024-bit ECFP4 | 0.7829 +- 0.0235 | 0.8883 +- 0.0230 | [Mufei Li]([email protected]) | [Paper](https://www.stat.berkeley.edu/~breiman/randomforest2001.pdf), [Code](./examples) | Jan 22nd, 2021 |
8 changes: 8 additions & 0 deletions examples/configures/GCN_GC/ClinTox.json
Original file line number Diff line number Diff line change
@@ -0,0 +1,8 @@
{
"batchnorm": true,
"dropout": 0.04333497108612183,
"hidden_feats": 128,
"lr": 0.15302291932022413,
"num_gnn_layers": 2,
"residual": true
}
6 changes: 6 additions & 0 deletions examples/configures/RF_ECFP/ClinTox.json
Original file line number Diff line number Diff line change
@@ -0,0 +1,6 @@
{
"bootstrap": false,
"criterion": "entropy",
"min_samples_split": 16,
"n_estimators": 100
}

0 comments on commit 2cc0328

Please sign in to comment.