sklearn中的predict与predict_proba的区别

假定在一个k分类问题中,测试集中共有n个样本。则:

  • predict返回的是一个大小为n的一维数组,一维数组中的第i个值为模型预测第i个预测样本的标签;
  • predict_proba返回的是一个n行k列的数组,第i行第j列上的数值是模型预测第i个预测样本的标签为j的概率。此时每一行的和应该等于1。
posted @ 2019-03-20 15:38  王小薇  阅读(990)  评论(0编辑  收藏  举报