sklearn中的predict與predict_proba的區別


假定在一個k分類問題中,測試集中共有n個樣本。則:

  • predict返回的是一個大小為n的一維數組,一維數組中的第i個值為模型預測第i個預測樣本的標簽;
  • predict_proba返回的是一個n行k列的數組,第i行第j列上的數值是模型預測第i個預測樣本的標簽為j的概率。此時每一行的和應該等於1。


免責聲明!

本站轉載的文章為個人學習借鑒使用,本站對版權不負任何法律責任。如果侵犯了您的隱私權益,請聯系本站郵箱yoyou2525@163.com刪除。



 
粵ICP備18138465號   © 2018-2025 CODEPRJ.COM