Keras中 predict（）与predict_classes()的区别

首先，大家要明白一点，Keras框架能使用的函数是predict和predict_classes；Sklearn中能使用的函数是predict和predict_proba。

 

以下是Keras中使用的predict和predict_ classes：

1.predict函数：训练后返回一个概率值数组，此数组的大小为 n·k，第i行第j列上对应的数值代表模型对此样本属于某类标签的概率值，行和为1。例如预测结果为：[[0.66651809 0.53348191]，代表预测样本的标签是0的概率为0.66651809，1的概率为0.53348191。

2.predict_ classes函数：返回一个标签数组，例如[3],代表样本大概率属于3号标签。