Fix mapping of predicted probabilities for class membership
Fix mapping of predicted probabilities for class membership

ceholden/yatsm,jmorton/yatsm,jmorton/yatsm,valpasq/yatsm,c11/yatsm,c11/yatsm,ceholden/yatsm,valpasq/yatsm,jmorton/yatsm