antoniaebner commited on
Commit
f1835be
·
1 Parent(s): 05ef690

update readme

Browse files
Files changed (1) hide show
  1. README.md +15 -0
README.md CHANGED
@@ -70,6 +70,21 @@ conda activate tox21_rf_cls
70
  pip install -r requirements.txt
71
  ```
72
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
73
  # Inference
74
 
75
  For inference, you only need `predict.py`.
 
70
  pip install -r requirements.txt
71
  ```
72
 
73
+ # Training
74
+
75
+ To train the Random Forest model from scratch, run:
76
+
77
+ ```bash
78
+ python preprocess.py
79
+ python train.py
80
+ ```
81
+
82
+ These commands will:
83
+ 1. Load and preprocess the Tox21 training dataset
84
+ 2. Train a Random Forest classifier
85
+ 3. Store the resulting model in the `checkpoints/` directory.
86
+
87
+
88
  # Inference
89
 
90
  For inference, you only need `predict.py`.