基于非连续数据的护林员预测森林

Question

我正在创建一个随机森林模型，在R中使用游侠，对MNIST像素数据进行训练，并预测像素所代表的数目(0-9)。在使用下面的代码创建我的模型并查看预测之后，我的许多预测数字都是连续的(小数)。

set.seed(1031)
forestranger <- ranger(Y ~ ., data = mnisttrain, num.trees = 1000)
predictions <- predict(forestranger, data = mnisttest, num.trees = 1000)

有没有办法确保我的预测只是0到9之间的整数，而不是连续的？

Answer 1

将响应变量转换为因子似乎解决了这个问题。