包使用: 'xgboost'版本0.4-4
我正在使用模型构建函数xgboost(),使用代码:
fit <- xgboost(data =sparse_matrix , label = trainSet$OutputClass,
max.depth = 4,eta = 1, nthread = 2, nround = 10,
eval_metric = "merror",objective = "multi:softmax",num_class = 45)
当我使用预测函数时:
Prediction<- predict(fit,sparse_matrixtestSet)
上面的代码给出了如下输出(而不是类名,它给出了数值等价值,尽管&#34; label = trainSet $ OutputClass&#34;包含类名)
输出:
[1] 1 1 1 1 1 35 3 3 3 4 31 7 7 7 3 3 9 9 9 9 9 9 9 10 10 11
[27] 11 11 11 11 11 11 11 11 11 13 13 13 13 13 13 13 13 13 14 14 14 14 14 14 10 10
[53] 15 15 15 15 15 15 15 15 15 15 15 16 16 16 16 16 16 16 16 16 16 16 16 18 18 18
[79] 18 18 18 18 35 35 35 18 21 21 21 21 32 1 1 25 25 25 25 26 27 27 27 27 27 27
[105] 27 27 29 29 29 29 29 30 30 30 30 30 30 30 30 30 30 35 35 32 32 32 43 43 32 32
[131] 32 32 32 32 32 32 43 32 32 32 32 32 33
我在阅读数据集时也设置了stringsAsFactors=FALSE
。
有人可以帮助我如何根据类名而不是数值来获取预测值... 提前致谢