Skip to content
Merged
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
add axis=1 to argmax call in logreg docs to fit new Theano behavior
  • Loading branch information
fbreuer committed Jan 11, 2013
commit 940cb40f9adf6fbb8f00f8bf1010b8d96362e76e
2 changes: 1 addition & 1 deletion doc/logreg.txt
Original file line number Diff line number Diff line change
Expand Up @@ -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)
Expand Down