diff --git a/modules/ml-pipeline/src/pipeline/configs/build_model.yaml b/modules/ml-pipeline/src/pipeline/configs/build_model.yaml
index 69b1c72..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: 180
+      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 923f3e0..04ab745 100644
--- a/modules/ml-pipeline/src/pipeline/dvc.lock
+++ b/modules/ml-pipeline/src/pipeline/dvc.lock
@@ -49,8 +49,8 @@ stages:
     outs:
     - path: data/prepared_data/
       hash: md5
-      md5: 2c85f5a6d81478de4efcb11c0f421e69.dir
-      size: 36926186
+      md5: 4cec69f112537658f14eb3cb678f91e3.dir
+      size: 36889932
       nfiles: 2
   build_model:
     cmd: python 2_build_model.py
@@ -61,8 +61,8 @@ stages:
       size: 4820
     - path: data/prepared_data
       hash: md5
-      md5: 2c85f5a6d81478de4efcb11c0f421e69.dir
-      size: 36926186
+      md5: 4cec69f112537658f14eb3cb678f91e3.dir
+      size: 36889932
       nfiles: 2
     params:
       configs/build_model.yaml:
@@ -79,7 +79,7 @@ stages:
               output_filepath: ./data/model/allmodels/
               problem_type: regression
               eval_metric: mean_squared_error
-              time_limit: 180
+              time_limit: 4000
               presets: medium_quality
               excluded_model_types:
               - RF
@@ -93,18 +93,18 @@ stages:
     outs:
     - path: data/fit_predictions/
       hash: md5
-      md5: 73cf2636e0272fc40c7540cb0975c649.dir
-      size: 2902177
+      md5: 7dda2f1dd257a6c5beaaa0b74eab6d5d.dir
+      size: 2901760
       nfiles: 1
     - path: data/model/
       hash: md5
-      md5: c6cdcfebd5dcdcc653bb2224f82170a8.dir
-      size: 341328895
-      nfiles: 24
+      md5: 741f8aed57383e860c535feb8b0adb71.dir
+      size: 752079341
+      nfiles: 32
     - path: metrics/fit_metrics.json
       hash: md5
-      md5: af7a36acbb7b216502afabaf846b6114
-      size: 215
+      md5: 8eaa72b08074f735a9e54de871edc6e6
+      size: 221
   generate_predictions:
     cmd: python 3_generate_predictions.py
     deps:
@@ -114,13 +114,13 @@ stages:
       size: 2464
     - path: data/model
       hash: md5
-      md5: c6cdcfebd5dcdcc653bb2224f82170a8.dir
-      size: 341328895
-      nfiles: 24
+      md5: 741f8aed57383e860c535feb8b0adb71.dir
+      size: 752079341
+      nfiles: 32
     - path: data/prepared_data
       hash: md5
-      md5: 2c85f5a6d81478de4efcb11c0f421e69.dir
-      size: 36926186
+      md5: 4cec69f112537658f14eb3cb678f91e3.dir
+      size: 36889932
       nfiles: 2
     params:
       configs/settings.yaml:
@@ -132,8 +132,8 @@ stages:
     outs:
     - path: data/predictions/
       hash: md5
-      md5: c0b6e1ae7a85f476e27926e041b76960.dir
-      size: 380477
+      md5: d842fe5350a3330c4c17e7e21c6359b2.dir
+      size: 380489
       nfiles: 1
   generate_metrics:
     cmd: python 4_generate_metrics.py
@@ -144,13 +144,13 @@ stages:
       size: 3447
     - path: data/predictions
       hash: md5
-      md5: c0b6e1ae7a85f476e27926e041b76960.dir
-      size: 380477
+      md5: d842fe5350a3330c4c17e7e21c6359b2.dir
+      size: 380489
       nfiles: 1
     - path: data/prepared_data
       hash: md5
-      md5: 2c85f5a6d81478de4efcb11c0f421e69.dir
-      size: 36926186
+      md5: 4cec69f112537658f14eb3cb678f91e3.dir
+      size: 36889932
       nfiles: 2
     params:
       configs/settings.yaml:
@@ -160,5 +160,5 @@ stages:
     outs:
     - path: metrics/metrics.json
       hash: md5
-      md5: 47a1fd7ba1b9eddeb6598ee6f2d06efb
-      size: 217
+      md5: 2632fa5d0a38763c177bf0466a670c8b
+      size: 220