将Binary类矩阵转换回类向量,并获得概率最高的前n个预测

时间:2019-03-19 17:52:05

标签: r tensorflow keras

在R Keras, 有一个函数使用函数to_categorical将类向量(整数)转换为二进制类矩阵。完成此操作后,我们通常将y_labels放入模型中,一旦获得结果,就可以使用predict()函数或predict_classes()函数进行预测。当前,predict_classes返回最可能的类预测。有没有一种方法可以根据概率返回前两个预测类别?例如,各种类别为1,2,3。

模型预测新数据将是10%3级,90%1级和0%2级,因此输出为:1 3

0 个答案:

没有答案