Skip to content

Commit 400c637

Browse files
committed
Merge pull request lisa-lab#15 from fbreuer/master
Fix argmax call in logreg tutorial docs to include axis=1.
2 parents 90a1bed + 940cb40 commit 400c637

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

doc/logreg.txt

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -83,7 +83,7 @@ The code to do this in Theano is the following:
8383

8484
# symbolic description of how to compute prediction as class whose probability
8585
# is maximal
86-
y_pred = T.argmax(p_y_given_x)
86+
y_pred = T.argmax(p_y_given_x, axis=1)
8787

8888
# compiled theano function that returns this value
8989
classify = theano.function(inputs=[x], outputs=y_pred)

0 commit comments

Comments
 (0)