Added file code_prediction, where code creating file out.tsv which results is saving to folders dev-0 and test-A

This commit is contained in:
Bartusiak 2020-03-29 00:58:08 +01:00
parent 4951a7cc47
commit 082c69e025

View File

@ -7,6 +7,7 @@ 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():
for token, tokens in word_logprobs[i].items():
if (word_logprobs['skeptic'][token] > word_logprobs['paranormal'][token]):