Skip to content


Subversion checkout URL

You can clone with
Download ZIP
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
Showing with 1 addition and 1 deletion.
  1. +1 −1  doc/logreg.txt
2  doc/logreg.txt
@@ -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.
Something went wrong with that request. Please try again.