def test_cond_prop(self):
hmm = HMM(UDDataSet("data/en-ud-train.conllu"))
cprob = hmm.cond_prob(4, 3, 7)
self.assertAlmostEqual(1.00, sum([np.exp(i) for i in cprob]))
often = hmm.dataset.lookup_word("often")
adj = hmm.dataset.lookup_pos("ADJ")
propn = hmm.dataset.lookup_pos("PROPN")
cprob = hmm.cond_prob(often, adj, propn)
self.assertAlmostEqual(1.00, sum([np.exp(i) for i in cprob]))
def test_error_cond_prop(self):
hmm = HMM(UDDataSet("data/en-ud-train.conllu"))
cprob = hmm.cond_prob(121, 4, 16)
self.assertAlmostEqual(1.00, sum([np.exp(i) for i in cprob]))