def test_bigram_smooth_lm_log_vecs():
intrp_lambda = 0.1
a = 1.
b = 2.
K = 5.
lm = BigramSmoothLM(intrp_lambda, a, b, K)
data = [
[1, 1, 3, 4, 0],
[4, 4],
[1, 0, 2, 2, 2, 2, 3, 1],
[3, 3, 1]
]
lm.counts_from_data(data)
log_prob_vec_i = lm.log_prob_vec_i()
for i in range(5):
npt.assert_almost_equal(log_prob_vec_i[i], np.log(lm.prob_i(i)))
j = 3
log_prob_vec_given_j = lm.log_prob_vec_given_j(j)
for i in range(5):
npt.assert_almost_equal(log_prob_vec_given_j[i], np.log(lm.prob_i_given_j(i, j)))
评论列表
文章目录