gklambauer commited on
Commit
d71474f
·
verified ·
1 Parent(s): 60782a2

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +6 -1
README.md CHANGED
@@ -13,7 +13,12 @@ short_description: This is a RF classifier for the Tox21 test dataset
13
 
14
  This repository hosts a Hugging Face Space that provides an examplary API for submitting models to the [Tox21 Leaderboard](https://huggingface.co/spaces/tschouis/tox21_leaderboard).
15
 
16
- In this example, we train a Random Forest classifier on the Tox21 targets and save the trained model in the `assets/` folder.
 
 
 
 
 
17
 
18
  **Important:** For leaderboard submission, your Space does not need to include training code. It only needs to implement inference in the `predict()` function inside `predict.py`. The `predict()` function must keep the provided skeleton: it should take a list of SMILES strings as input and return a prediction dictionary as output, with SMILES and targets as keys. Therefore, any preprocessing of SMILES strings must be executed on-the-fly during inference.
19
 
 
13
 
14
  This repository hosts a Hugging Face Space that provides an examplary API for submitting models to the [Tox21 Leaderboard](https://huggingface.co/spaces/tschouis/tox21_leaderboard).
15
 
16
+
17
+ Here **Random Forest (RF)** models are trained on the Tox21 dataset, and the trained models are provided for
18
+ inference. For each of the twelve toxic effects, a separate RF model is trained. The input to the model
19
+ is a **SMILES** string of the small molecule, and the output are 12 numeric values for
20
+ each of the toxic effects of the Tox21 dataset.
21
+
22
 
23
  **Important:** For leaderboard submission, your Space does not need to include training code. It only needs to implement inference in the `predict()` function inside `predict.py`. The `predict()` function must keep the provided skeleton: it should take a list of SMILES strings as input and return a prediction dictionary as output, with SMILES and targets as keys. Therefore, any preprocessing of SMILES strings must be executed on-the-fly during inference.
24