def test_train_labels_no_labels(self):
layer_sizes = [5, 20, 10]
d = DBN(layer_sizes, 2)
data = [[1, 1, 1, 1, 1], [1, 1, 1, 1, 1]]
labels = []
with self.assertRaises(ValueError):
d.train_labels(data, labels, 1, 1)
def test_train_labels_no_labels(self):
layer_sizes = [5, 20, 10]
d = DBN(layer_sizes, 2)
data = [[1, 1, 1, 1, 1], [1, 1, 1, 1, 1]]
labels = []
with self.assertRaises(ValueError):
d.train_labels(data, labels, 1, 1)
def test_train_labels(self):
layer_sizes = [5, 20, 10]
d = DBN(layer_sizes, 2)
data = [[1, 1, 1, 1, 1], [1, 1, 1, 1, 1]]
labels = [[1, 0], [0, 1]]
d.train_labels(data, labels, 1, 1)
def test_train_labels(self):
layer_sizes = [5, 20, 10]
d = DBN(layer_sizes, 2)
data = [[1, 1, 1, 1, 1], [1, 1, 1, 1, 1]]
labels = [[1, 0], [0, 1]]
d.train_labels(data, labels, 1, 1)
