poprawki

stupidsolution.py

@@ -1,17 +1,22 @@
 import csv
-
+import re
 
 def makeoutput(infile, outfile):
-    counter = 0
+    output = []
+    regex = r'paranormal|ufo|youtube|spirit'
     with open(infile) as tsvfile:
         reader = csv.reader(tsvfile, delimiter='\t')
         for row in reader:
-            counter+=1
+            if re.search(regex, str(row).lower()):
+                output.append('1')
+            else:
+                output.append('0')
+
 
     with open(outfile, 'wt') as tsvfile:
         tsv_writer = csv.writer(tsvfile, delimiter='\t')
-        for i in range(counter):
-            tsv_writer.writerow('S')
+        for i in output:
+            tsv_writer.writerow(i)
 
 makeoutput("test-A/in.tsv", "test-A/out.tsv")
 makeoutput("train/in.tsv", "train/out.tsv")