def test_product1(self): got = self._compute_until(self.data, "dense_product_1") expected = product1_out(sum0_out(product0_out(normal_leafs(self.data)))) self.assertAllClose(got, expected)
def test_root(self): got = tf.exp(self.spn(self.data)) expected = root_out(product1_out(sum0_out(product0_out(normal_leafs(self.data))))) self.assertAllClose(got, expected)
def test_normal_leafs(self): got = self._compute_until(self.data, "normal_leaf") expected = normal_leafs(self.data) self.assertAllClose(got, expected)