如何将朴素贝叶斯的预测输出转换为数据帧对象

Question

当我运行预测函数时，我会以

的形式获得多个类的输出

class a , class b , class c , class d
[1,] 0.007883349 0.001882366 0.0004159344 0.003825768 0.0001649481 0.0108488 0.005848486
 class e , class f , class g
[1,] 0.4 0.003

我无法将它们转换为数据帧，因为我需要找到前3个类。

Answer 1

我终于得到了如何以数据帧格式呈现：通过将预测结果以矩阵格式放置，然后使用转置将其转换为数据帧

rw <- as.matrix(predict(model,usr_input,type='raw'))
cc <- as.data.frame(t(rw))