Ejemplo n.º 1
0
    def test_kldiv(self):
        p = np.array([[0.1, 0.4, 0.3, 0.9]])
        q1 = np.array([[0.1, 0.2, 0.2, 0.5]])
        q2 = np.array([[0.1, 0.4, 0.3, 0.8]])

        self.assertTrue(TopicModel.kldiv(p, p) == 0)

        d1 = TopicModel.kldiv(p, q1)
        self.assertGreater(d1, 0)
        d2 = TopicModel.kldiv(p, q2)
        self.assertGreater(d1, d2)