fix predict

dnouri · May 26, 2016 · b2bd1e5 · b2bd1e5
1 parent 1047acc
commit b2bd1e5
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/nolearn/lasagne/base.py b/nolearn/lasagne/base.py
@@ -672,7 +672,9 @@ def predict(self, X, y=None):
  if self.regression:
  return self.predict_proba(X, y)
  else:
- predictions, y_actual = self.predict_proba(X, y)
+ predictions = self.predict_proba(X, y)
+ if y is not None:
+ predictions, y_actual = predictions
  y_pred = np.argmax(predictions, axis=1)
  if self.use_label_encoder:
  y_pred = self.enc_.inverse_transform(y_pred)