dvc ref

2022-06-05 17:01:04 +02:00 · 2022-06-05 17:01:04 +02:00 · 8216b1705f
commit 8216b1705f
parent 4dd3799dbb
2 changed files with 1 additions and 5 deletions
--- a/data_test.csv.dvc
+++ b/data_test.csv.dvc
@ -1,4 +0,0 @@
-outs:
- md5: b57fd92c7032152f66ce5d169c2c3fdc
-  size: 6994695
-  path: data_test.csv
--- a/evaluation.py
+++ b/evaluation.py
@ -17,7 +17,7 @@ if __name__ == "__main__":
    company_profile = vectorizer.fit_transform(company_profile)
    model = torch.load('model')

-    data_test = pd.read_csv('data_test.csv', engine='python', header=None)
+    data_test = data[1:100]
    data_test = data_test.dropna()
    x_test = data_test[5]
    y_test = data_test[17]