4000 model

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

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