staged.

Chunk
1 parent 0bd44a28
Showing 1 changed file with 1 additions and 1 deletions Show diff stats
mmodel/theano/THEANO.py
@@ -17,7 +17,7 @@ class ModelTHEANO(ModelBase):
         self.sparker = sc
         self.model = None
  
-    def _train_cnn(X, Y, learning_rate=0.1, n_epochs=200, nkerns=[20, 50, 50],
+    def _train_cnn(self, X, Y, learning_rate=0.1, n_epochs=200, nkerns=[20, 50, 50],
                    batch_size=200):
         X_train, X_test, Y_train, Y_test = cross_validation.train_test_split(X, Y, test_size=0.2, random_state=0)
...	...	@@ -17,7 +17,7 @@ class ModelTHEANO(ModelBase):
17	17	self.sparker = sc
18	18	self.model = None
19	19
20		- def _train_cnn(X, Y, learning_rate=0.1, n_epochs=200, nkerns=[20, 50, 50],
	20	+ def _train_cnn(self, X, Y, learning_rate=0.1, n_epochs=200, nkerns=[20, 50, 50],
21	21	batch_size=200):
22	22	X_train, X_test, Y_train, Y_test = cross_validation.train_test_split(X, Y, test_size=0.2, random_state=0)
23	23
...	...