poprawki

solution.py

@@ -64,8 +64,8 @@ def calc_words_logprobs(words_counts):
     total_paranormal = sum(word_counts['paranormal'].values() + len(word_counts['paranormal'].keys()))
 
 
-# with open('prediction.tsv', 'wt') as tsvfile:
-#     tsv_writer = csv.writer(tsvfile, delimiter='\t')
-#     for i in range(counter):
-#         tsv_writer.writerow('S')
+with open('prediction.tsv', 'wt') as tsvfile:
+    tsv_writer = csv.writer(tsvfile, delimiter='\t')
+    for i in range(counter):
+        tsv_writer.writerow('S')
 

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")