-
Notifications
You must be signed in to change notification settings - Fork 1
/
dvc.yaml
104 lines (98 loc) · 3.31 KB
/
dvc.yaml
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
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
stages:
prepare_datasets:
cmd: python3 src/flare_classifier/prepare_datasets.py data/light-curves/positive_demo.csv
data/light-curves/negative_demo.csv
deps:
- src/flare_classifier/prepare_datasets.py
- data/light-curves/positive_demo.csv
- data/light-curves/negative_demo.csv
params:
- prepare_datasets.train_size
- prepare_datasets.test_size
- prepare_datasets.seed
outs:
- data/prepared
train_rf:
cmd: python3 src/flare_classifier/train_rf.py data/prepared
params:
- train_rf.n_estimators
- train_rf.seed
deps:
- src/flare_classifier/train_rf.py
- src/flare_classifier/prepare_datasets.py
- data/prepared/train.parquet
- data/prepared/feature_names.csv
outs:
- models/random_forest/random_forest.pickle
train_catboost:
cmd: python3 src/flare_classifier/train_catboost.py data/prepared/
deps:
- src/flare_classifier/train_catboost.py
- src/flare_classifier/prepare_datasets.py
- data/prepared/train.parquet
- data/prepared/feature_names.csv
params:
- train_catboost.depth
- train_catboost.iterations
- train_catboost.learning_rate
- train_catboost.loss_function
- train_catboost.seed
outs:
- models/catboost/catboost.pickle
train_mlp:
cmd: python3 src/flare_classifier/train_mlp.py data/prepared/
deps:
- src/flare_classifier/train_mlp.py
- src/flare_classifier/prepare_datasets.py
- src/flare_classifier/mlp_model.py
- data/prepared/train.parquet
- data/prepared/val.parquet
- data/prepared/feature_names.csv
params:
- train_mlp.lr
- train_mlp.batch_size
- train_mlp.num_epochs
outs:
- models/mlp/mlp.pickle
evaluate_rf:
cmd: python3 src/flare_classifier/evaluate_rf.py models/random_forest/random_forest.pickle
data/prepared/ data/light-curves/real_flares_test.csv
deps:
- src/flare_classifier/metrics.py
- src/flare_classifier/train_rf.py
- src/flare_classifier/evaluate_rf.py
- data/prepared/train.parquet
- data/prepared/test.parquet
- data/light-curves/real_flares_test.csv
- models/random_forest/random_forest.pickle
outs:
- metrics/random_forest/train_metrics.json
- metrics/random_forest/test_metrics.json
evaluate_catboost:
cmd: python3 src/flare_classifier/evaluate_catboost.py models/catboost/catboost.pickle
data/prepared/ data/light-curves/real_flares_test.csv
deps:
- src/flare_classifier/metrics.py
- src/flare_classifier/train_catboost.py
- src/flare_classifier/evaluate_catboost.py
- data/prepared/train.parquet
- data/prepared/test.parquet
- data/light-curves/real_flares_test.csv
- models/catboost/catboost.pickle
outs:
- metrics/catboost/train_metrics.json
- metrics/catboost/test_metrics.json
evaluate_mlp:
cmd: python3 src/flare_classifier/evaluate_mlp.py models/mlp/mlp.pickle
data/prepared/ data/light-curves/real_flares_test.csv
deps:
- src/flare_classifier/metrics.py
- src/flare_classifier/train_mlp.py
- src/flare_classifier/evaluate_mlp.py
- data/prepared/train.parquet
- data/prepared/test.parquet
- data/light-curves/real_flares_test.csv
- models/mlp/mlp.pickle
outs:
- metrics/mlp/train_metrics.json
- metrics/mlp/test_metrics.json