First commit with setup and DVC files

.dvc/.gitignore

@@ -0,0 +1,3 @@
+/config.local
+/tmp
+/cache

.dvc/config

@@ -0,0 +1,5 @@
+[core]
+    analytics = false
+    remote = remote_storage
+['remote "remote_storage"']
+    url = /home/mlops/dvc_remote

.dvcignore

@@ -0,0 +1,3 @@
+# Add patterns of files dvc should ignore, which could improve
+# the performance. Learn more at
+# https://dvc.org/doc/user-guide/dvcignore

data/prepared/.gitignore

@@ -0,0 +1,2 @@
+/train.csv
+/test.csv

data/prepared/test.csv.dvc

@@ -0,0 +1,4 @@
+outs:
+- md5: cbd4ba69ced15e40820a635e7e741627
+  size: 71491
+  path: test.csv

data/prepared/train.csv.dvc

@@ -0,0 +1,4 @@
+outs:
+- md5: 7f79cf9a4ab1316f7d7246d0b92ea16c
+  size: 178060
+  path: train.csv

data/raw/.gitignore

@@ -0,0 +1,2 @@
+/train
+/val

data/raw/train.dvc

@@ -0,0 +1,5 @@
+outs:
+- md5: 7adc7abb69056f4d7afb512c78f2fce9.dir
+  size: 75309082
+  nfiles: 9470
+  path: train

data/raw/val.dvc

@@ -0,0 +1,5 @@
+outs:
+- md5: 0ad4dcf197b452735726bf8d8777201d.dir
+  size: 31248080
+  nfiles: 3925
+  path: val

metrics/accuracy.json

@@ -0,0 +1 @@
+{"accuracy": 0.7490494296577946}

model/.gitignore

@@ -0,0 +1 @@
+/model.joblib

model/model.joblib.dvc

@@ -0,0 +1,4 @@
+outs:
+- md5: 22490d1b369e3f7423c5b6ebd4db4234
+  size: 241075
+  path: model.joblib

src/train.py

@@ -37,7 +37,7 @@ def load_data(data_path):
 def main(repo_path):
     train_csv_path = repo_path / "data/prepared/train.csv"
     train_data, labels = load_data(train_csv_path)
-    sgd = SGDClassifier(max_iter=10)
+    sgd = SGDClassifier(max_iter=100)
     trained_model = sgd.fit(train_data, labels)
     dump(trained_model, repo_path / "model/model.joblib")