From 778bff37fb7808518cf570b321ab00f7bccd7726 Mon Sep 17 00:00:00 2001 From: Michael Duong Date: Fri, 9 Feb 2024 18:46:19 +0000 Subject: [PATCH] 4000 model --- .../src/pipeline/configs/build_model.yaml | 2 +- modules/ml-pipeline/src/pipeline/dvc.lock | 46 +++++++++---------- 2 files changed, 24 insertions(+), 24 deletions(-) diff --git a/modules/ml-pipeline/src/pipeline/configs/build_model.yaml b/modules/ml-pipeline/src/pipeline/configs/build_model.yaml index f4770a7..fcec7f7 100644 --- a/modules/ml-pipeline/src/pipeline/configs/build_model.yaml +++ b/modules/ml-pipeline/src/pipeline/configs/build_model.yaml @@ -14,7 +14,7 @@ default: output_filepath: ./data/model/allmodels/ problem_type: regression eval_metric: mean_squared_error #mean_absolute_error - time_limit: 400 + time_limit: 4000 presets: medium_quality excluded_model_types: ['RF', 'FASTAI', 'CAT', 'NN_TORCH', 'KNN', 'XT'] infer_limit: 0.05 diff --git a/modules/ml-pipeline/src/pipeline/dvc.lock b/modules/ml-pipeline/src/pipeline/dvc.lock index 9bb73b8..69729e0 100644 --- a/modules/ml-pipeline/src/pipeline/dvc.lock +++ b/modules/ml-pipeline/src/pipeline/dvc.lock @@ -31,8 +31,8 @@ stages: outs: - path: data/prepared_data/ hash: md5 - md5: f85c36a5dfd31a897538b3934d5fb997.dir - size: 41375196 + md5: cfaebbb77306750fcc9a39adbb40015b.dir + size: 41367957 nfiles: 2 build_model: cmd: python 2_build_model.py @@ -43,8 +43,8 @@ stages: size: 4820 - path: data/prepared_data hash: md5 - md5: f85c36a5dfd31a897538b3934d5fb997.dir - size: 41375196 + md5: cfaebbb77306750fcc9a39adbb40015b.dir + size: 41367957 nfiles: 2 params: configs/build_model.yaml: @@ -61,7 +61,7 @@ stages: output_filepath: ./data/model/allmodels/ problem_type: regression eval_metric: mean_squared_error - time_limit: 400 + time_limit: 4000 presets: medium_quality excluded_model_types: - RF @@ -75,17 +75,17 @@ stages: outs: - path: data/fit_predictions/ hash: md5 - md5: 991e6c55826953aa7c2be573369ec96f.dir - size: 3574047 + md5: f15cbb7486924de81c5bf032b5ca962d.dir + size: 3572461 nfiles: 1 - path: data/model/ hash: md5 - md5: f8a8b7462831bd46b1e2df47d73bb69d.dir - size: 391430703 - nfiles: 23 + md5: 5d41efafd16cda31e10a0ca1a0a19759.dir + size: 798325885 + nfiles: 31 - path: metrics/fit_metrics.json hash: md5 - md5: 35a66a845854cc6fee9dd10860e216bb + md5: 7790bc6b081a2c933547f67c843fef10 size: 225 generate_predictions: cmd: python 3_generate_predictions.py @@ -96,13 +96,13 @@ stages: size: 2464 - path: data/model hash: md5 - md5: f8a8b7462831bd46b1e2df47d73bb69d.dir - size: 391430703 - nfiles: 23 + md5: 5d41efafd16cda31e10a0ca1a0a19759.dir + size: 798325885 + nfiles: 31 - path: data/prepared_data hash: md5 - md5: f85c36a5dfd31a897538b3934d5fb997.dir - size: 41375196 + md5: cfaebbb77306750fcc9a39adbb40015b.dir + size: 41367957 nfiles: 2 params: configs/settings.yaml: @@ -114,8 +114,8 @@ stages: outs: - path: data/predictions/ hash: md5 - md5: 94b7381ac318b1ca18e0bc086778f7ce.dir - size: 626160 + md5: 0f476eaeaca81fbc7fac9400c77f653a.dir + size: 626833 nfiles: 1 generate_metrics: cmd: python 4_generate_metrics.py @@ -126,13 +126,13 @@ stages: size: 3484 - path: data/predictions hash: md5 - md5: 94b7381ac318b1ca18e0bc086778f7ce.dir - size: 626160 + md5: 0f476eaeaca81fbc7fac9400c77f653a.dir + size: 626833 nfiles: 1 - path: data/prepared_data hash: md5 - md5: f85c36a5dfd31a897538b3934d5fb997.dir - size: 41375196 + md5: cfaebbb77306750fcc9a39adbb40015b.dir + size: 41367957 nfiles: 2 params: configs/settings.yaml: @@ -142,7 +142,7 @@ stages: outs: - path: metrics/metrics.json hash: md5 - md5: 4d8681f7c0f41f97be52d6b1ae039c5b + md5: e000a99df92dc0c7ca86b020cbcd6b5b size: 224 startup_cleanup: cmd: python 0_startup_cleanup.py