Функция sklearn predict_proba().
Подходит для решающего дерева,случайного леса, логистической регрессии.
count_probabilities = model.predict_proba(df)