示例#1
0
 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)
示例#2
0
 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)
示例#3
0
 def train_rbm(self):
     r = RBM(5, 10)
     data = [[1,1,1,1,1],[0,0,0,0,0]]
     r.train(data,1,1)