dvc

.dvc/.gitignore

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

.dvc/config

@@ -0,0 +1,4 @@
+[core]
+    remote = ium_ssh_remote
+['remote "ium_ssh_remote"']
+    url = ssh://ium-sftp@tzietkiewicz.vm.wmi.amu.edu.pl

.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

.gitignore

@@ -14,3 +14,5 @@ ipython_config.py
 
 __pycache__/
 
+/train
+/test

dvc.yaml

@@ -0,0 +1,23 @@
+stages:
+  prepare_data:
+    cmd: ./download.sh
+    deps:
+      - download.sh
+    outs:
+      - data/raw
+
+  train:
+    cmd: python train.py
+    deps:
+      - train.py
+      - data/raw
+    outs:
+      - model.pth
+
+  test:
+    cmd: python test.py
+    deps:
+      - test.py
+      - model.pth
+    outs:
+      - predictions.csv

test.dvc

@@ -0,0 +1,6 @@
+outs:
+- md5: 31361b67b19276a528020b964be5a880.dir
+  size: 11777108
+  nfiles: 2000
+  hash: md5
+  path: test

train.dvc

@@ -0,0 +1,6 @@
+outs:
+- md5: 27bf2e3326fa79daa24727ffd1a9889f.dir
+  size: 71490454
+  nfiles: 11879
+  hash: md5
+  path: train