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,7 +17,7 @@ class ModelTHEANO(ModelBase):
17	self.sparker = sc	17	self.sparker = sc
18	self.model = None	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	batch_size=200):	21	batch_size=200):
22	X_train, X_test, Y_train, Y_test = cross_validation.train_test_split(X, Y, test_size=0.2, random_state=0)	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