标签: 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