paranormal-or-skeptic/code_prediction.py

11 lines
405 B
Python
Raw Normal View History

2020-03-28 20:40:28 +01:00
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'])