@@ -1,40 +1,56 @@
-featurization:
-  cmd: python code/featurization.py
-  deps:
-  - path: code/featurization.py
-    md5: ddf63559873ad5aa352565bb3b76bf0b
-  - path: data/test_data.csv
-    md5: 9686a73874063a8814cb95a7349887ad
-  - path: data/train_data.csv
-    md5: 5b49cf1b57fb9d6102b559d59d99df7c
-  outs:
-  - path: data/norm_params.json
-    md5: 9157137d0957f485519a620767b5a70f
-  - path: data/processed_test_data.npy
-    md5: 9bdbf1af9efdf2f6c4a56a1e15c2c847
-  - path: data/processed_train_data.npy
-    md5: 816c8566b0e07275f85133cc14987052
-training:
-  cmd: python code/train_model.py
-  deps:
-  - path: code/train_model.py
-    md5: 655c3242c17b3d0213d7ce4d9f78344d
-  - path: data/processed_train_data.npy
-    md5: 816c8566b0e07275f85133cc14987052
-  outs:
-  - path: data/model.pkl
-    md5: e854edcda9528d7ffae12c0fd4e423a7
-  - path: metrics/train_metric.json
-    md5: 46e9072b0a0105aeaacb76b736517b00
-eval:
-  cmd: python code/eval.py
-  deps:
-  - path: code/eval.py
-    md5: 75c6e9b5ffa5ab5bbf83b7c3e8de188d
-  - path: data/model.pkl
-    md5: e854edcda9528d7ffae12c0fd4e423a7
-  - path: data/processed_test_data.npy
-    md5: 9bdbf1af9efdf2f6c4a56a1e15c2c847
-  outs:
-  - path: metrics/eval.json
-    md5: ea8cda4deecfecaaa742c9d2bd6368ef
+schema: '2.0'
+stages:
+  featurization:
+    cmd: python code/featurization.py
+    deps:
+    - path: code/featurization.py
+      md5: ddf63559873ad5aa352565bb3b76bf0b
+      size: 1472
+    - path: data/test_data.csv
+      md5: c807df8d6d804ab2647fc15c3d40f543
+      size: 18289443
+    - path: data/train_data.csv
+      md5: c807df8d6d804ab2647fc15c3d40f543
+      size: 18289443
+    outs:
+    - path: data/norm_params.json
+      md5: 855b3da318f0dfd5eb36d4765092473a
+      size: 138861
+    - path: data/processed_test_data.npy
+      md5: 48b7c5120556c80e0c2c08602b53a216
+      size: 1280128
+    - path: data/processed_train_data.npy
+      md5: 48b7c5120556c80e0c2c08602b53a216
+      size: 1280128
+  training:
+    cmd: python code/train_model.py
+    deps:
+    - path: code/train_model.py
+      md5: 655c3242c17b3d0213d7ce4d9f78344d
+      size: 1652
+    - path: data/processed_train_data.npy
+      md5: 48b7c5120556c80e0c2c08602b53a216
+      size: 1280128
+    outs:
+    - path: data/model.pkl
+      md5: ed938ddb3939b89a5c05c7cdc1887f48
+      size: 740490
+    - path: metrics/train_metric.json
+      md5: 84dd09d6a04f5fb2fee433b936b0a4dd
+      size: 36
+  eval:
+    cmd: python code/eval.py
+    deps:
+    - path: code/eval.py
+      md5: 75c6e9b5ffa5ab5bbf83b7c3e8de188d
+      size: 974
+    - path: data/model.pkl
+      md5: ed938ddb3939b89a5c05c7cdc1887f48
+      size: 740490
+    - path: data/processed_test_data.npy
+      md5: 48b7c5120556c80e0c2c08602b53a216
+      size: 1280128
+    outs:
+    - path: metrics/eval.json
+      md5: aef91ad4f382159925ea132a024806e9
+      size: 20