BoTZ-TND
/
datastorm-2021
mirror of https://github.com/BoTZ-TND/datastorm-2021.git


  
1

	
2

	
3

	
4

	
5

	
6

	
7

	
8

	
9

	
10

	
11

	
12

	
13

	
14

	
15

	
16

	
17

	
18

	
19

	
20

	
21

	
22

	
23

	
24

	
25

	
26

	
27

	
28

	
29

	
30

	
31

	
32

	
33

	
34

	
35

	
36

	
37

	
38

	
39

	
40

	
41

	
42

	
43

	
44

	
45

	
46

	
47

	
48

	
49

	
50

	
51

	
52

	
53

	
54

	
55

	
56

	
57

	
58

	
59

	
            feature:
  cmd: python src/models/Cat-Boost/featurize_adv.py data/data data/feature
  deps:
  - path: data/data
    md5: e7bca3c665cd3995619c17283791527b.dir
    size: 5070940
    nfiles: 5
  - path: src/models/Cat-Boost/featurize.py
    md5: 5af2ae8135ddbd01977b0d7327f3374e
    size: 6508
  params:
    params.yaml:
      featurize.fs_select: 0
  outs:
  - path: data/feature
    md5: 7abd1c36287849cf7ee322157ccd81a1.dir
    size: 11016170
    nfiles: 10
train:
  cmd: python src/models/Cat-Boost/train.py data/feature/cba_train.csv
  deps:
  - path: data/feature/cb_train.csv
    md5: fb4415a34d42cbb1b00d118bde169ea9
    size: 1311004
  params:
    params.yaml:
      train.device: GPU
      train.f1-type: macro
      train.loss_function: MultiClass
      train.n_iter: 100
      train.seed: 443
      train.train_size: 0.9
  outs:
  - path: parameters.json
    md5: 3f79b19a72a5b1f79d0d2e6728963aaa
    size: 193
evaluation:
  cmd: python src/models/kitchen_sink/evaluate.py data/feature data/model/xgb-cl.pkl
  deps:
  - path: data/feature
    md5: 65f139576689e1cbb31623f641189df8.dir
    size: 2847333
    nfiles: 7
  - path: src/models/kitchen_sink/parameters.json
    md5: 3713a1c0481349a0c8b683db04b0d1e4
    size: 94
  params:
    params.yaml:
      train.lr: 0.01
      train.n_estimators: 1000
      train.objective: multi:softmax
  outs:
  - path: data/model/xgb-cl.pkl
    md5: 88208e18bd608ca95d57e01863926f47
    size: 1299523
  - path: score.json
    md5: ca897952db25125a3f4739b1f4ed7daa
    size: 276