Как визуализировать и распечатать правила модели classif.rpart из пакета mlr, как в rpart.plot и rpart.rules? - PullRequest
0 голосов
/ 13 февраля 2019

Я создаю правила из дерева решений, сгенерированного учеником-упаковщиком mlr classif.rpart, как распечатать правила как в rpart.rules и как визуализировать как в rpart.plot

Создал ученикаиспользуя classif.rpart, обучил и подогнал модель, попытался построить с помощью rpart.plot и сказал ошибку как Не объект rpart

dt_mod <- mlr::train(fused_dt, classif.task)
dt_mod
library(rpart.plot)
rpart.plot(dt_mod$learner.model, roundint=FALSE, varlen=3, type = 3, clip.right.labs = FALSE, yesno = 2)
rpart.rules(dt_mod$learner.model, roundint = FALSE)

I, за исключением правил, которые должны быть перечислены, и дерева, которое нужно визуализировать

1 Ответ

0 голосов
/ 13 февраля 2019

Вы можете получить доступ к модели, созданной учеником напрямую, с помощью getLearnerModel():

iris.model = train(makeLearner("classif.rpart"), iris.task)
rpart.plot(getLearnerModel(iris.model))

Если ваш ученик упакован, передайте опцию more.unwrap = TRUE:

rpart.plot(getLearnerModel(iris.model, more.unwrap = TRUE))
Добро пожаловать на сайт PullRequest, где вы можете задавать вопросы и получать ответы от других членов сообщества.
...