python - 如何在二元分类器中按预测概率进行排序？

Question

我训练了一个二元分类器并获得了不错的分数。

reviews['prediction'] = model.predict(reviews.review.astype(str))
model.score(reviews.review.astype(str), reviews.sentiment)
model.predict_proba(reviews.review.astype(str))

当我打印predict_prob时，我得到了数组中的概率

array([[ 0. ,  1. ],
       [ 0. ,  1. ],
       [ 0.7,  0.3],
       ..., 
       [ 0. ,  1. ],
       [ 0. ,  1. ],
       [ 0. ,  1. ]])

如何获取最高概率并将其附加到数据框，就像我对类所做的那样？

Answer 1

可以对数组调用max(axis=1)来获取每行的最大值，例如:

reviews['proba'] = predict_prob.max(axis=1)