leonicus
/
dagshub_project


  
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

	
            schema: '2.0'
stages:
  preprocessing:
    cmd: python code/preprocess.py
    deps:
    - path: code/preprocess.py
      md5: d4c3885065dbf70be7d172ad44ade8d9
      size: 1512
    - path: data/X_train.csv
      md5: d85e637571665773be2c1093fa2ac250
      size: 281577207
    - path: data/y_train.json
      md5: c5149d885b896f58e7347438081fbc6d
      size: 14768000
    outs:
    - path: processed_data/X_test.npy
      md5: 94eb9ab5130417f7a6bc299a0e61d73d
      size: 49179234
    - path: processed_data/X_train_processed.npy
      md5: 831e8833e4527470d9ab5f61e0b3fc95
      size: 67250518
    - path: processed_data/X_val_processed.npy
      md5: 051ca6cd097df25563f9fb0f1fac47c0
      size: 24266587
    - path: processed_data/y_train_processed.npy
      md5: 35c1fee22c8faee2a42bf4e9fc469290
      size: 12658408
    - path: processed_data/y_val_processed.npy
      md5: 219acae9e050ec81e65c4a3ac5d22890
      size: 4219560
  training:
    cmd: python code/train_model.py
    deps:
    - path: code/train_model.py
      md5: ed31e76602b7a1cc6894aa3244121462
      size: 1294
    - path: processed_data/X_train_processed.npy
      md5: 831e8833e4527470d9ab5f61e0b3fc95
      size: 67250518
    - path: processed_data/X_val_processed.npy
      md5: 051ca6cd097df25563f9fb0f1fac47c0
      size: 24266587
    - path: processed_data/y_train_processed.npy
      md5: 35c1fee22c8faee2a42bf4e9fc469290
      size: 12658408
    - path: processed_data/y_val_processed.npy
      md5: 219acae9e050ec81e65c4a3ac5d22890
      size: 4219560
    outs:
    - path: trained_models/catboost.pk1
      md5: 8b378e26b66dae784946aa180a3f6c52
      size: 593183029