def test_classify_wrong_shape(self): layer_sizes = [5, 20, 10] d = DBN(layer_sizes, 2) data = [[1, 1, 1, 1]] with self.assertRaises(ValueError): probs = d.classify(data)
def test_classify(self): layer_sizes = [5, 20, 10] d = DBN(layer_sizes, 2) data = [[1, 1, 1, 1, 1]] probs = d.classify(data)