We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
2 parents 90a1bed + 940cb40 commit 400c637Copy full SHA for 400c637
doc/logreg.txt
@@ -83,7 +83,7 @@ The code to do this in Theano is the following:
83
84
# symbolic description of how to compute prediction as class whose probability
85
# is maximal
86
- y_pred = T.argmax(p_y_given_x)
+ y_pred = T.argmax(p_y_given_x, axis=1)
87
88
# compiled theano function that returns this value
89
classify = theano.function(inputs=[x], outputs=y_pred)
0 commit comments