diff --git a/code_prediction.py b/code_prediction.py index 454cab0..870459c 100644 --- a/code_prediction.py +++ b/code_prediction.py @@ -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]):