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)