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_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)
def test_classify(self):
layer_sizes = [5, 20, 10]
d = DBN(layer_sizes, 2)
data = [[1, 1, 1, 1, 1]]
probs = d.classify(data)
