-
Notifications
You must be signed in to change notification settings - Fork 0
/
alt-pipeline.yaml
81 lines (73 loc) · 2.48 KB
/
alt-pipeline.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
meta:
# import_tasks_from: features.yaml
# Extract upstream dependencies from source code. If False, tasks
# must declare dependencies using the "upstream" key
extract_upstream: False
# Extract product from source code. If False, tasks must have a "product" key
extract_product: False
executor: serial
tasks:
- source: func.util.data_transform
name: data-transform
upstream: null
product: products/data/data_alt.csv
params:
file_path: '{{root}}/{{DATA_FILEPATH}}'
- source: scripts/features.py
name: features-alt
upstream: [ data-transform ]
product:
data: products/data/features_odds_alt.csv
nb: products/reports/features_odds_alt.ipynb
params:
features: '{{features_odds_alt_all}}'
target: '{{TARGET_ALT}}'
- source: scripts/nested_cv_alt.py
name: nested-cv-alt
upstream: [ features-alt ]
product:
nb: products/reports/nested_cv_alt.ipynb
model: products/models/nested_cv_alt.pt
params:
target: '{{TARGET_ALT}}'
odds_cols: '{{ODDS_COLS_ALL}}'
random_seed: '{{RANDOM_SEED}}'
cat_features: '{{CAT_FEATURES}}'
test_ratio: 0
inner_splits: 2
outer_splits: 5
# - source: func.util.split_train_test_op_alt
# name: split-train-test-alt
# upstream: [ features-alt ]
# product:
# train: products/data/train_alt.csv
# test: products/data/test_alt.csv
# train_odds: products/data/train_odds_alt.csv
# test_odds: products/data/test_odds_alt.csv
# params:
# test_ratio: '{{TEST_RATIO}}'
# - source: scripts/fit_pytorch_alt.py
# name: fit-pytorch-alt
# upstream: [ split-train-test-alt ]
# product:
# nb: products/reports/fit_pytorch_alt.ipynb
# model_state_dict: products/models/pytorch_state_dict_alt.pt
# model: products/models/pytorch_alt.pt
# params:
# target: '{{TARGET_ALT}}'
# pytorch_conf: '{{pytorch_conf}}'
# random_seed: '{{RANDOM_SEED}}'
# validation_ratio: 0.2
# odds_cols: ''
#
# - source: scripts/eval_pytorch_alt.py
# name: eval-pytorch-alt
# upstream: [ split-train-test-alt, fit-pytorch-alt ]
# product:
# nb: products/reports/eval_pytorch_alt.ipynb
# params:
# target: '{{TARGET_ALT}}'
# pytorch_conf: '{{pytorch_conf}}'
# odds_cols: '{{odds_cols_alt}}'
# bootstrap_repetitions: 3
# kelly_fraction: 0.05