From 8216b1705f0aeb8fb16cd74e31cb25ad1f590c01 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Miko=C5=82aj=20Pokrywka?= <mp7961@gmail.com>
Date: Sun, 5 Jun 2022 17:01:04 +0200
Subject: [PATCH] dvc ref

---
 data_test.csv.dvc | 4 ----
 evaluation.py     | 2 +-
 2 files changed, 1 insertion(+), 5 deletions(-)
 delete mode 100644 data_test.csv.dvc

diff --git a/data_test.csv.dvc b/data_test.csv.dvc
deleted file mode 100644
index b3ee600..0000000
--- a/data_test.csv.dvc
+++ /dev/null
@@ -1,4 +0,0 @@
-outs:
-- md5: b57fd92c7032152f66ce5d169c2c3fdc
-  size: 6994695
-  path: data_test.csv
diff --git a/evaluation.py b/evaluation.py
index bcace13..d4fa7dc 100644
--- 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]