paranormal-or-skeptic/code_prediction.py
2020-03-28 20:40:28 +01:00

11 lines
405 B
Python

from collections import defaultdict
import math
import pickle
open_file=open('naive_base_model.pkl','rb')
pickle_loaded=pickle.load(open_file)
paranomal_class_logprob, skeptic_class_logprob, word_logprobs = pickle_loaded
opened_file=open('test.tsv',"w")
#print(paranomal_class_logprob, skeptic_class_logprob, word_logprobs)
print(word_logprobs['skeptic']['ufo'])
print(word_logprobs['paranormal']['ufo'])