2017-09-14 3 views
1

Il y a une fonction dans le paquet r mlr qui liste toutes les méthodes qu'il supporte pour un apprenant donné, que j'ai utilisé une fois mais que je ne retrouve pas. Je me souviens que xgb.create.feature de xgboost était définitivement sur la liste incluse, mais je ne peux trouver aucun document sur la façon de l'utiliser à partir de mlr. Est-ce que quelqu'un sait comment faire ça? (Et si quelqu'un peut se rappeler le nom de la fonction de recherche mlr pour les méthodes d'apprentissage implémentées qui serait également très apprécié.)mlr support pour xgb.create.features - comment utiliser?

+0

Un grand merci. Cela fonctionne parfaitement. – andy

Répondre

2

xgb.create.features est une fonction de xgboost pas mlr.

Si vous souhaitez utiliser la fonction, vous pouvez accéder directement au modèle de l'apprenant et appeler la fonction.

library(mlr) 
library(xgboost) 

mod = train(makeLearner("classif.xgboost"), iris.task) 
iris.dc = data.matrix(getTaskData(iris.task, target.extra = TRUE)$data) 
xgboost::xgb.create.features(mod$learner.model, iris.dc) 

Toutes les méthodes d'apprenants ne sont pas directement supportées par le côté mlr.