Skip to content

Commit 01ea8bb

Browse files
committed
Run experiments tuning random forest params
1 parent 153e68c commit 01ea8bb

File tree

2 files changed

+10
-10
lines changed

2 files changed

+10
-10
lines changed

dvc.lock

Lines changed: 8 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -57,14 +57,14 @@ stages:
5757
size: 1666
5858
params:
5959
params.yaml:
60-
train.min_split: 0.01
61-
train.n_est: 50
60+
train.min_split: 8
61+
train.n_est: 100
6262
train.seed: 20170428
6363
outs:
6464
- path: model.pkl
6565
hash: md5
66-
md5: d1f6e055f7f5e2827fcfae68d9b64d4c
67-
size: 1958115
66+
md5: 63cd44fd46b7f76875dc9556a2000cb5
67+
size: 10322743
6868
evaluate:
6969
cmd: python src/evaluate.py model.pkl data/features
7070
deps:
@@ -75,15 +75,15 @@ stages:
7575
nfiles: 2
7676
- path: model.pkl
7777
hash: md5
78-
md5: d1f6e055f7f5e2827fcfae68d9b64d4c
79-
size: 1958115
78+
md5: 63cd44fd46b7f76875dc9556a2000cb5
79+
size: 10322743
8080
- path: src/evaluate.py
8181
hash: md5
8282
md5: a1a59f55636170fb56e0c6afd3e28fa4
8383
size: 3315
8484
outs:
8585
- path: eval
8686
hash: md5
87-
md5: 843f7816d0f99e3b1fd1717c54856dc4.dir
88-
size: 1292360
87+
md5: bc55587cf8d870b9f861907fe0864d68.dir
88+
size: 1186900
8989
nfiles: 8

params.yaml

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -8,6 +8,6 @@ featurize:
88

99
train:
1010
seed: 20170428
11-
n_est: 50
12-
min_split: 0.01
11+
n_est: 100
12+
min_split: 8
1313

0 commit comments

Comments
 (0)