Skip to content

Commit 000628d

Browse files
authored
Update README.md
1 parent 81df748 commit 000628d

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

README.md

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -51,6 +51,7 @@ python3 -m chebai predict_from_file --model=[path-to-model-config] --checkpoint_
5151
```
5252
The input files should contain a list of line-separated SMILES strings. This generates a CSV file that contains the
5353
one row for each SMILES string and one column for each class.
54+
The `classes_path` is the path to the dataset's `raw/classes.txt` file that contains the relationship between model output and ChEBI-IDs.
5455

5556
## Evaluation
5657

@@ -68,4 +69,4 @@ and the fold to be used in the current optimisation run as
6869
--data.init_args.fold_index=I
6970
```
7071
To train K models, you need to do K such calls, each with a different `fold_index`. On the first call with a given
71-
`inner_k_folds`, all folds will be created and stored in the data directory
72+
`inner_k_folds`, all folds will be created and stored in the data directory

0 commit comments

Comments
 (0)