paranormal-or-skeptic/code_prediction.py

from collections import defaultdict
import math
import pickle

open_file=open('naive_base_model.pkl','rb')
write_file_test=open('test-A/out.tsv','w')
write_file_dev=open('dev-0/out.tsv','w')
pickle_loaded=pickle.load(open_file)
paranomal_class_logprob, skeptic_class_logprob, word_logprobs = pickle_loaded
#Niektórych słów nie bezie w zbiorze treningowym dev-0 i dev-A
for i in word_logprobs.keys():
    try:
        for token, tokens in word_logprobs[i].items():
            if (word_logprobs['skeptic'][token] > word_logprobs['paranormal'][token]):
                write_file_test.write("S\n")
                write_file_dev.write("S\n")
            else:
                write_file_test.write("P\n")
                write_file_dev.write("P\n")
    except KeyError:
        print(token + " not avaible")