ISI-33, probabilities

solution.py

@@ -25,18 +25,18 @@ testA = pd.read_csv("test-A/in.tsv", delimiter="\t", header=None, names=["text",
 dev0_new_counts = count_vect.transform(dev0)
 testA_new_counts = count_vect.transform(testA)
 
-predicted_dev0 = clf.predict(dev0_new_counts)
-predicted_testA = clf.predict(testA_new_counts)
+predicted_dev0 = clf.predict_proba(dev0_new_counts)
+predicted_testA = clf.predict_proba(testA_new_counts)
 
 print(len(dev0))
 print(len(predicted_dev0))
 
 with open("dev-0/out.tsv", "w") as out1:
     for line in predicted_dev0:
-        out1.write(str(line))
+        out1.write(str(line[1]))
         out1.write("\n")
 
 with open("test-A/out.tsv", "w") as out2:
     for line in predicted_testA:
-        out2.write(str(line))
+        out2.write(str(line[1]))
         out2.write("\n")