首页 > 解决方案 > 来自 autoxgboost 模型的特征重要性

问题描述

我有使用 autoxgboost 自动调整的模型,它也使用 mlr,请您告诉我如何在训练模型上使用 getFeatureImportance?

system.time(tuned.labeled.autoxgboost <- autoxgboost(labeled_task))
preds.labeled.autoxgboost <- predict(tuned.labeled.autoxgboost$final.model, newdata = labeled_test)
class(tuned.labeled.autoxgboost)
[1] "AutoxgbResult"
class(tuned.labeled.autoxgboost$final.model)
[1] "CPOModel" "BaseWrapperModel" "WrappedModel"
getFeatureImportance(tuned.labeled.autoxgboost$final.model)
Error in UseMethod("getFeatureImportanceLearner") :
no applicable method for 'getFeatureImportanceLearner' applied to an object of class "c('regr.xgboost.custom', 'RLearnerRegr', 'RLearner', 'Learner')"
sapply(tuned.labeled.autoxgboost$final.model,class)
$learner
[1] "CPOLearner" "BaseWrapper" "Learner"

$learner.model
[1] "CPOWrappedModel" "ChainModel" "WrappedModel"

$task.desc
[1] "RegrTaskDesc" "SupervisedTaskDesc" "TaskDesc"

$subset
[1] "integer"

$features
[1] "character"

$factor.levels
[1] "list"

$time
[1] "numeric"

$dump
[1] "NULL"

我是否应该使用优化的参数在 mlr 中设置 xgboost 学习器,然后运行 ​​getFeatureImportance?

谢谢你。

标签: rmlr

解决方案


推荐阅读