You might also consider the decision models, d1-2B is very strong and good for calibrated classification. Would be happy to share links/resources if you're interested.