def test_trans_prob_2gram(self):
hmm = MLHMM(2, self.tagged_sents, addone=False)
probs = {
('D', ('<s>',)): 1.0,
('N', ('D',)): 1.0,
('V', ('N',)): 0.5,
('N', ('V',)): 1.0,
('P', ('N',)): 0.5,
('</s>', ('P',)): 1.0,
}
for params, p in probs.items():
self.assertAlmostEqual(hmm.trans_prob(*params), p, msg=params)
def test_trans_prob_2gram(self):
hmm = MLHMM(2, self.tagged_sents, addone=False)
probs = {
('D', ('<s>', )): 1.0,
('N', ('D', )): 1.0,
('V', ('N', )): 0.5,
('N', ('V', )): 1.0,
('P', ('N', )): 0.5,
('</s>', ('P', )): 1.0,
}
for params, p in probs.items():
self.assertAlmostEqual(hmm.trans_prob(*params), p, msg=params)
