Probability estimation in pymc3