diff --git a/modules/ml-pipeline/src/pipeline/configs/build_model.yaml b/modules/ml-pipeline/src/pipeline/configs/build_model.yaml index c8c022d..49fbac4 100644 --- a/modules/ml-pipeline/src/pipeline/configs/build_model.yaml +++ b/modules/ml-pipeline/src/pipeline/configs/build_model.yaml @@ -13,6 +13,6 @@ default: output_filepath: ./data/model/autogluonmodel/ problem_type: regression eval_metric: mean_absolute_error - time_limit: 75 + time_limit: 1000 presets: medium_quality excluded_model_types: ['KNN'] diff --git a/modules/ml-pipeline/src/pipeline/dvc.lock b/modules/ml-pipeline/src/pipeline/dvc.lock index aeb705d..9b9a3e0 100644 --- a/modules/ml-pipeline/src/pipeline/dvc.lock +++ b/modules/ml-pipeline/src/pipeline/dvc.lock @@ -30,8 +30,8 @@ stages: outs: - path: data/prepared_data/ hash: md5 - md5: ed19a11a85d6a2006631173f51569d27.dir - size: 21131576 + md5: f9ef7ad073b43b249b43faa75c62fe07.dir + size: 21115444 nfiles: 2 build_model: cmd: python 2_build_model.py @@ -42,8 +42,8 @@ stages: size: 5181 - path: data/prepared_data hash: md5 - md5: ed19a11a85d6a2006631173f51569d27.dir - size: 21131576 + md5: f9ef7ad073b43b249b43faa75c62fe07.dir + size: 21115444 nfiles: 2 params: configs/build_model.yaml: @@ -59,20 +59,20 @@ stages: output_filepath: ./data/model/autogluonmodel/ problem_type: regression eval_metric: mean_absolute_error - time_limit: 75 + time_limit: 1000 presets: medium_quality excluded_model_types: - KNN outs: - path: data/model/ hash: md5 - md5: 60e253c42cc36934098c627ef3ef4cc1.dir - size: 185134993 - nfiles: 14 + md5: d073af40ba5c7c2d9b8064665062f51e.dir + size: 363710367 + nfiles: 20 - path: metrics/fit_metrics.json hash: md5 - md5: a0c2a1c9e5da0b857d510fa1ba6282a8 - size: 186 + md5: dcd9ea03a2771077e1bd14018bb7fd18 + size: 183 generate_predictions: cmd: python 3_generate_predictions.py deps: @@ -82,13 +82,13 @@ stages: size: 4720 - path: data/model hash: md5 - md5: 60e253c42cc36934098c627ef3ef4cc1.dir - size: 185134993 - nfiles: 14 + md5: d073af40ba5c7c2d9b8064665062f51e.dir + size: 363710367 + nfiles: 20 - path: data/prepared_data hash: md5 - md5: ed19a11a85d6a2006631173f51569d27.dir - size: 21131576 + md5: f9ef7ad073b43b249b43faa75c62fe07.dir + size: 21115444 nfiles: 2 params: configs/settings.yaml: @@ -100,8 +100,8 @@ stages: outs: - path: data/predictions/ hash: md5 - md5: 700c8767de3a86c4c5339daf3cc17192.dir - size: 380962 + md5: a2ecfae1e418fe9cb9fe044c148bbb37.dir + size: 381538 nfiles: 1 generate_metrics: cmd: python 4_generate_metrics.py @@ -112,13 +112,13 @@ stages: size: 4487 - path: data/predictions hash: md5 - md5: 700c8767de3a86c4c5339daf3cc17192.dir - size: 380962 + md5: a2ecfae1e418fe9cb9fe044c148bbb37.dir + size: 381538 nfiles: 1 - path: data/prepared_data hash: md5 - md5: ed19a11a85d6a2006631173f51569d27.dir - size: 21131576 + md5: f9ef7ad073b43b249b43faa75c62fe07.dir + size: 21115444 nfiles: 2 params: configs/settings.yaml: @@ -128,8 +128,8 @@ stages: outs: - path: metrics/metrics.json hash: md5 - md5: 45ffac8f6e7283df4b69af8a9abc45e1 - size: 184 + md5: ec02774fd01243fa4706189c60087ccf + size: 182 startup_cleanup: cmd: python 0_startup_cleanup.py deps: