Multinomial logistic regression in pymc3