Probability estimation in pymc3

Yes. When you are doing binary classification, then the mean value(axis=0) would be the class probability.

Things are a bit more complicated if you are doing multi-class classification using categorical distribution (eg., observed = pm.Categorical('obs', p=p, observed=yt)). In those cases, you need to find a way to sample the node p.

1 Like