如何按二元分類器中的預測概率進行排序？

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)