Improve model

Author: janecummings

model.pkl.dvc

@@ -4,17 +4,17 @@ deps:
   path: data/splitter/train.csv
 - md5: c38869db7776de4a4298934196ce2991
   path: data/splitter/validation.csv
-- md5: 7186e47534570e956e1c72a0154fa5e5
+- md5: ca25619e8be91531afa2ad7da1734b5f
   path: src/decision_tree.py
-md5: 999c895ec54438c45952f2f54e58ada2
+md5: 88b713b21c3caa7dbacf9a282b90be05
 outs:
 - cache: true
-  md5: 1594dc237f58879bdeba2040cd8ddc7e
+  md5: 26a1a260bf9e822b56d0d07da1f5d59a
   metric: false
   path: data/decision_tree/model.pkl
   persist: false
 - cache: false
-  md5: 4461d85c84d1cff7fa84f6503c1fd374
+  md5: ba0b144d94a938360a0c42cac86d7703
   metric: true
   path: results/metrics.json
   persist: false

results/metrics.json

@@ -1 +1 @@
-{"nwrmsle": 0.8261571666321248, "r2_score": -0.8594341254951023}
+{"nwrmsle": 0.743416398731014, "r2_score": 0.10865649241119235}