completed train_model

.gitignore

@@ -89,3 +89,6 @@ target/
 .mypy_cache/
 /models
 /data
+
+
+test.py

data.dvc

@@ -1,6 +1,6 @@
 outs:
-- md5: 67d4b2315b7329a2fd73d5e43e1b2fb4.dir
-  size: 677683647
+- md5: 88caf4a67eea0b88a2f3048a7ee35cd1.dir
+  size: 504622770
   nfiles: 9
   hash: md5
   path: data

dvc.lock

@@ -20,5 +20,74 @@ stages:
     deps:
     - path: .\src\features\build_features.py
       hash: md5
-      md5: c2ae958800f9aaf01f39a0b9833b5276
-      size: 2287
+      md5: 867164fc77fb07a349206e3ab682eb41
+      size: 2523
+  train_model:
+    cmd: python  .\src\models\train_model.py .\data\processed\
+    params:
+      params.yaml:
+        train_model.alpha:
+        - 1.0
+        - 0.5
+        - 1.5
+        - 0.8
+        train_model.bagging_n_estimators:
+        - 500
+        - 100
+        - 250
+        train_model.base_estimator:
+        - LinearRegression()
+        - Ridge()
+        - Lasso()
+        - DecisionTreeRegressor()
+        train_model.criterion:
+        - squared_error
+        - friedman_mse
+        - absolute_error
+        - poisson
+        train_model.criterion_g:
+        - squared_error
+        - friedman_mse
+        train_model.l1_ratio:
+        - 0.5
+        - 0.8
+        - 0.2
+        train_model.learning_rate:
+        - 0.1
+        - 0.01
+        - 0.5
+        - 0.001
+        train_model.loss:
+        - linear
+        - square
+        - exponential
+        train_model.loss_g:
+        - squared_error
+        - absolute_error
+        - huber
+        - quantile
+        train_model.max_features:
+        - 0.5
+        - 0.25
+        train_model.max_iter:
+        - 1000
+        - 800
+        - 900
+        - 1500
+        train_model.max_samples:
+        - 0.25
+        - 0.5
+        train_model.n_estimators:
+        - 8
+        - 64
+        - 128
+        - 256
+        - 512
+        train_model.splitter:
+        - best
+        - random
+        train_model.subsample:
+        - 0.6
+        - 0.75
+        - 0.8
+        - 0.9

dvc.yaml

@@ -10,4 +10,22 @@ stages:
   build_features:
     cmd: python .\src\features\build_features.py .\data\interim\train.csv .\data\interim\test.csv
     deps:
-    - .\src\features\build_features.py
+    - .\src\features\build_features.py
+  train_model:
+    cmd: python  .\src\models\train_model.py .\data\processed\
+    params:
+    - train_model.alpha
+    - train_model.max_iter
+    - train_model.l1_ratio
+    - train_model.criterion
+    - train_model.splitter
+    - train_model.max_features
+    - train_model.learning_rate
+    - train_model.n_estimators
+    - train_model.loss_g
+    - train_model.loss
+    - train_model.subsample
+    - train_model.criterion_g
+    - train_model.base_estimator
+    - train_model.bagging_n_estimators
+    - train_model.max_samples