sonja-a-topf commited on
Commit
a391b68
·
verified ·
1 Parent(s): 976e441

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +12 -0
README.md CHANGED
@@ -70,6 +70,18 @@ cd tox21_chemprop_classifier
70
  pip install -r requirements.txt
71
  ```
72
 
 
 
 
 
 
 
 
 
 
 
 
 
73
 
74
  # Inference
75
 
 
70
  pip install -r requirements.txt
71
  ```
72
 
73
+ # Training
74
+
75
+ To train the Chemprop model from scratch, run:
76
+
77
+ ```bash
78
+ python train.py
79
+ ```
80
+
81
+ These commands will:
82
+ 1. Load and preprocess the Tox21 training dataset
83
+ 2. Train a Chemprop classifier
84
+ 3. Store the resulting model in the `checkpoints/` directory.
85
 
86
  # Inference
87