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