[Sklearn]-模型.predict和模型.predict_proba的区别

595 阅读1分钟

区别

  • 假定测试集数据是60*(5+1),共有2个类别。
方法功能返回值类型返回值shape
predict()返回标签值<class 'numpy.ndarray'>(60,)
predict_proba()返回的是一个 60 行 2 列的数组, 第 i 行 第 j 列上的数值是模型预测 第 i 个预测样本为某个标签的概率,并且每一行的概率和为1<class 'numpy.ndarray'>(60, 2)