[英]Predict() with sklearn LogisticRegression and RandomForest models always predict minority class (1)
[英]probability missing in sklearn LogisticRegression.predict
我正在嘗試將數據分類為3類之一。 我正在使用sklearns logistic回歸,該回歸將輸入單詞到我的測試數據的vec向量化形式中作為輸入。 預測方法返回預測,但是沒有給出有關模型在分類正確的概率或置信度的數字? 有人可以告訴我如何獲取該信息嗎?
如果您仔細閱讀文檔 ,則還有另一個函數predict_proba
來獲得概率。
因此,您只需要執行以下操作:
clf = LogisticRegression()
clf.predict_proba(X_input)
聲明:本站的技術帖子網頁,遵循CC BY-SA 4.0協議,如果您需要轉載,請注明本站網址或者原文地址。任何問題請咨詢:yoyou2525@163.com.