GHIPrediction/dvc.yaml at main · se4ai2324-uniba/GHIPrediction · GitHub

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
stages:
  merging:
    cmd: python3 src/data/make_dataset.py
    deps:
    - data/raw/2017.csv
    - data/raw/2018.csv
    - data/raw/2019.csv
    - src/data/make_dataset.py
    outs:
    - data/raw/FinaleBari.csv

  preprocessing:
    cmd: python3 src/data/preprocessing.py
    deps:
    - data/raw/FinaleBari.csv
    - src/data/preprocessing.py
    outs:
    - data/processed/GHI.csv
    - data/processed/PreprocessedData.csv
  split_dataset:
    cmd: python3 src/data/split_dataset.py
    deps:
    - data/processed/GHI.csv
    - data/processed/PreprocessedData.csv
    - src/data/split_dataset.py
    outs:
    - data/interim/split_train.csv
    - data/interim/split_test.csv
  Knr:
    cmd: python3 src/models/knr.py
    deps:
    - data/interim/split_train.csv
    - data/interim/split_test.csv
    - src/models/knr.py
    - src/models/train_model.py
    outs:
    - models/knr.pkl
    metrics:
    - models/knr.metrics
  XGBooster:
    cmd: python3 src/models/xgbooster.py
    deps:
    - data/interim/split_train.csv
    - data/interim/split_test.csv
    - src/models/xgbooster.py
    - src/models/train_model.py
    outs:
    - models/xgb.pkl
    metrics:
    - models/xgb.metrics
  LinearRegression:
    cmd: python3 src/models/linear_regressor.py
    deps:
    - data/interim/split_train.csv
    - data/interim/split_test.csv
    - src/models/linear_regressor.py
    - src/models/train_model.py
    outs:
    - models/linear.pkl
    metrics:
    - models/linear.metrics
  RandomForestRegressor:
    cmd: python3 src/models/random_forest_regressor.py
    deps:
    - data/interim/split_train.csv
    - data/interim/split_test.csv
    - src/models/random_forest_regressor.py
    - src/models/train_model.py
    outs:
    - models/randomForestRegressor.pkl
    metrics:
    - models/randomForestRegressor.metrics

  Compare:
    cmd: python3 src/models/compare.py
    deps:
    - models/randomForestRegressor.metrics
    - models/randomForestRegressor.pkl
    - models/linear.metrics
    - models/linear.pkl
    - models/xgb.metrics
    - models/xgb.pkl
    - models/knr.metrics
    - models/knr.pkl
    outs:
    - models/best_model.pkl
    metrics:
    - models/best_metrics.metrics