Permalink
Browse files

add axis=1 to argmax call in logreg docs to fit new Theano behavior

  • Loading branch information...
1 parent 90a1bed commit 940cb40f9adf6fbb8f00f8bf1010b8d96362e76e @fbreuer fbreuer committed Jan 11, 2013
Showing with 1 addition and 1 deletion.
  1. +1 −1 doc/logreg.txt
View
@@ -83,7 +83,7 @@ The code to do this in Theano is the following:
# symbolic description of how to compute prediction as class whose probability
# is maximal
- y_pred = T.argmax(p_y_given_x)
+ y_pred = T.argmax(p_y_given_x, axis=1)
# compiled theano function that returns this value
classify = theano.function(inputs=[x], outputs=y_pred)

0 comments on commit 940cb40

Please sign in to comment.