关于来自R的xgboost包的预测值

时间:2016-10-14 06:22:52

标签: r prediction xgboost

包使用: '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

有人可以帮助我如何根据类名而不是数值来获取预测值... 提前致谢

0 个答案:

没有答案