def test_train_rbm_wrong_shape(self): r = RBM(5, 10) data = [[0,0,0,0]] with self.assertRaises(ValueError): r.train(data,1,1)
def test_train_rbm_negative_batches(self): r = RBM(5, 10) data = [[1,1,1,1,1],[0,0,0,0,0]] with self.assertRaises(ValueError): r.train(data,1,-1)
def train_rbm(self): r = RBM(5, 10) data = [[1,1,1,1,1],[0,0,0,0,0]] r.train(data,1,1)