Skip to content

Commit 128e196

Browse files
Modify README
1 parent 9e4724c commit 128e196

File tree

3 files changed

+19
-1
lines changed

3 files changed

+19
-1
lines changed

README.md

Lines changed: 19 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -103,9 +103,27 @@ After running this script the following datasets are being generated:
103103
python steps/evaluate_baselines.py
104104
```
105105

106+
Offline metrics (AUROC & NDCG) of baselines solutions:
107+
108+
![alt text](static/evaluate_baselines.png)
109+
110+
### 3. Train
111+
112+
Training MLP based on user histories for 20 epochs:
113+
114+
```bash
115+
python steps/train.py experiment=mlp_with_history trainer.max_epochs=20
116+
```
117+
118+
![alt text](static/train.png)
119+
120+
### 4. Rest of the steps
121+
106122
```bash
107-
python steps/train.py
123+
python steps/optimize_hparams.py
108124
python steps/infer.py
125+
python steps/serve.py
126+
python steps/run_pipeline.py
109127
```
110128

111129
## Other useful commands:

static/evaluate_baselines.png

50.6 KB
Loading

static/train.png

68.2 KB
Loading

0 commit comments

Comments
 (0)