Annalie
/
star-types
mirror of https://github.com/annalieNK/star-types


  
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

	
            prepare:
  cmd: python src/prepare.py data/data.csv
  deps:
  - path: data/data.csv
    md5: 9501858625a46ac02f30ada4d4412355
    size: 8482
  - path: src/prepare.py
    md5: 14b9e15966ed3f56c3089b9788e6d9ec
    size: 1159
  outs:
  - path: data/prepared
    md5: 955e35dacb9a0cd8db7008809ed2e7ff.dir
    size: 13721
    nfiles: 1
train:
  cmd: python src/train.py data/prepared model.pkl
  deps:
  - path: data/prepared
    md5: 955e35dacb9a0cd8db7008809ed2e7ff.dir
    size: 13721
    nfiles: 1
  - path: src/train.py
    md5: a8fcd82b405f527747c66334046bca46
    size: 1963
  params:
    params.yaml:
      train.model: random forest
      train.seed: 1
      train.split: 0.2
  outs:
  - path: model/model.pkl
    md5: b01fb5014d5354f97230b22b0b8b28e6
    size: 266339
evaluate:
  cmd: python src/evaluate.py model.pkl data/prepared scores.json confusion_matrix.png
    ROC_AUC_curve.png
  deps:
  - path: data/prepared
    md5: 955e35dacb9a0cd8db7008809ed2e7ff.dir
    size: 13721
    nfiles: 1
  - path: model/model.pkl
    md5: b01fb5014d5354f97230b22b0b8b28e6
    size: 266339
  - path: src/evaluate.py
    md5: 181e1ae01d709dee6c311f55a2905516
    size: 2795
  params:
    params.yaml:
      evaluate.model: random forest
      evaluate.seed: 1
      evaluate.split: 0.2