Mercurial > pylearn
diff mlp.py @ 155:ae5651a3696b
new argmax calling convention
| author | James Bergstra <bergstrj@iro.umontreal.ca> |
|---|---|
| date | Mon, 12 May 2008 16:16:32 -0400 |
| parents | 3f4e5c9bdc5e |
| children | e9a95e19e6f8 4090779e39a9 |
line wrap: on
line diff
--- a/mlp.py Mon May 12 15:51:43 2008 -0400 +++ b/mlp.py Mon May 12 16:16:32 2008 -0400 @@ -89,7 +89,7 @@ self._regularization_term = self._L2_regularizer * (t.sum(self._W1*self._W1) + t.sum(self._W2*self._W2)) self._output_activations =self._b2+t.dot(t.tanh(self._b1+t.dot(self._input,self._W1.T)),self._W2.T) self._nll,self._output = crossentropy_softmax_1hot(self._output_activations,self._target_vector) - self._output_class, self._max_output = t.argmax(self._output,1) + self._output_class = t.argmax(self._output,1) self._class_error = t.neq(self._output_class,self._target_vector) self._minibatch_criterion = self._nll + self._regularization_term / t.shape(self._input)[0] OnlineGradientTLearner.__init__(self)