diff --git a/unsupervised_class2/autoencoder.py b/unsupervised_class2/autoencoder.py index 87633fa1..4e9b85eb 100644 --- a/unsupervised_class2/autoencoder.py +++ b/unsupervised_class2/autoencoder.py @@ -60,7 +60,7 @@ def fit(self, X, learning_rate=0.5, mu=0.99, epochs=1, batch_sz=100, show_fig=Fa self.dparams = [self.dW, self.dbh, self.dbo] self.forward_dparams = [self.dW, self.dbh] - X_in = T.matrix('X_%s' % self.id) + X_in = T.matrix('X_%s' % self.id, dtype='float32') X_hat = self.forward_output(X_in) # attach it to the object so it can be used later @@ -177,7 +177,7 @@ def fit(self, X, Y, Xtest, Ytest, for ae in self.hidden_layers: self.params += ae.forward_params - X_in = T.matrix('X_in') + X_in = T.matrix('X_in', dtype='float32') targets = T.ivector('Targets') pY = self.forward(X_in)