From 9e2ef895ff68634a2d44c8be81b16ca233dcbea7 Mon Sep 17 00:00:00 2001
From: Kacper Dudzic <s449288@pc19f.labs.wmi.amu.edu.pl>
Date: Fri, 29 Apr 2022 18:46:18 +0200
Subject: [PATCH] file loading fix

---
 run.py | 8 ++++++--
 1 file changed, 6 insertions(+), 2 deletions(-)

diff --git a/run.py b/run.py
index c45d631..76bd61f 100644
--- a/run.py
+++ b/run.py
@@ -7,10 +7,14 @@ train_file = sys.argv[1]
 pred_file = sys.argv[2]
 
 train = pd.read_csv(train_file, sep='\t', header=None)
-pred_x = pd.read_csv(pred_file, sep='\t', header=None)
+#pred_x = pd.read_csv(pred_file, sep='\t', header=None)
+pred_x = []
+with open(pred_file, encoding='utf-8') as f:
+    for line in f:
+        pred_x.append(line)
 train_x, train_y = train[4], train[0]
 #pred_x = pred[4]
-pred_x = pred_x.stack()
+#pred_x = pred_x.stack()
 
 vectorizer = TfidfVectorizer()
 train_x = vectorizer.fit_transform(train_x)