r - 重要性排名 : error must be an object of class xgb. 助推器

Question

我运行了一个 xgboost 回归预测(也尝试用 xgb.Booster.complete 完成它)。尝试获取 xgb.importance 时，我收到错误信息

Error in xgboost::xgb.importance(case_xgbm) : model: must be anobject of class xgb.Booster

但是，在验证时，R 说它是一个“xgb.Booster”类。知道发生了什么事吗？

library(xgboost)
library(caret)        
somedata <- MASS::Boston
    
    indexes = createDataPartition(somedata$medv, p = .85, list = F) #medv is the y
    train = somedata[indexes, ]
    test = somedata[-indexes, ]
    
    train_x = data.matrix(train[, -13])
    train_y = train[,13]
    
    xgb_train = xgb.DMatrix(data = train_x, label = train_y)
    xgbc = xgboost(data = xgb_train, max.depth = 2, nrounds = 50)
    class(xgbc)
    xgboost::xgb.importance(xgbc)
    
    xgbc2 = xgb.Booster.complete(xgbc, saveraw = TRUE)
    class(xgbc2)
    xgboost::xgb.importance(xgbc2)

Answer 1

尝试xgboost::xgb.importance(model=xgbc)这对我有用