Skip to content

Commit 6a38317

Browse files
committed
update readme for new prediction method
1 parent 7b7e48f commit 6a38317

File tree

1 file changed

+12
-4
lines changed

1 file changed

+12
-4
lines changed

README.md

Lines changed: 12 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -70,11 +70,19 @@ python -m chebai fit --config=[path-to-your-tox21-config] --trainer.callbacks=co
7070

7171
### Predicting classes given SMILES strings
7272
```
73-
python3 -m chebai predict_from_file --checkpoint_path=[path-to-model] --input_path=[path-to-file-containing-smiles] [--classes_path=[path-to-classes-file]] [--save_to=[path-to-output]]
73+
python3 chebai/result/prediction.py predict_from_file --checkpoint_path=[path-to-model] ----smiles_file_path=[path-to-file-containing-smiles] [--classes_path=[path-to-classes-file]] [--save_to=[path-to-output]]
7474
```
75-
The input files should contain a list of line-separated SMILES strings. This generates a CSV file that contains the
76-
one row for each SMILES string and one column for each class.
77-
The `classes_path` is the path to the dataset's `raw/classes.txt` file that contains the relationship between model output and ChEBI-IDs.
75+
76+
* **`--checkpoint_path`**: Path to the Lightning checkpoint file (must end with `.ckpt`).
77+
78+
* **`--smiles_file_path`**: Path to a text file containing one SMILES string per line.
79+
80+
* **`--save_to`** *(optional)*: Predictions will be saved to the path as CSV file. The CSV will contain one row per SMILES string and one column per predicted class.
81+
82+
* **`--classes_path`** *(optional)*: Path to the dataset’s `raw/classes.txt` file, which maps model output indices to ChEBI IDs.
83+
84+
* If provided, the CSV columns will be named using the ChEBI IDs.
85+
* If omitted, then script will located the file automatically. If unable to locate then the columns will be numbered sequentially.
7886

7987
## Evaluation
8088

0 commit comments

Comments
 (0)