Модель Caret, как обучить окончательную модель с использованием оптимизированных параметров?

train_control <- trainControl(method='cv', number=10)    
model <- train(Class ~ Age+BMI+DBP+DPF+NumPregnancies+PG2+SI2+TSFT, method ='rf',data=input,trControl=train_control)

pmml(model$finalModel)
#Error in names(field$class) <- var.names : attempt to set an attribute on NULL

Когда я пытаюсь экспортировать модель RF, обученную каретке, в PMML, это не удается. Есть ли способ запуска RF вручную с использованием оптимальных параметров, настроенных пакетом Caret, чтобы я мог экспортировать модель в PMML?


r r-caret pmml
person madawa    schedule 27.01.2015    source источник
comment
Какую именно ошибку вы получаете? Ваш пример невозможно воспроизвести, потому что tr_con не определено.   -  person MrFlick    schedule 27.01.2015
comment
Не могли бы вы привести воспроизводимый пример? Непонятно даже, какую библиотеку вы используете.   -  person emhart    schedule 27.01.2015
comment
stackoverflow.com/questions/27428748 / В ответе на этот вопрос говорится, что модель Caret rf не может быть экспортирована как pmml, и рекомендуется использовать следующий двухуровневый подход. Во-первых, используйте пакет Caret, чтобы найти наиболее подходящие РЧ-параметры для набора данных. Во-вторых, обучите окончательную радиочастотную модель вручную, используя интерфейс формулы с этими параметрами. Я хочу знать, как можно выполнить вторую часть этого.   -  person madawa    schedule 27.01.2015
comment
@DistribEcology Я использую библиотеку caret в R   -  person madawa    schedule 27.01.2015
comment
Вы загрузили пакет randomForest перед выполнением команды pmml?   -  person topepo    schedule 27.01.2015

Ответы (2)


arrow_upward
1
arrow_downward

Доступ к настроенным параметрам можно получить через model$bestTune

> model$bestTune
mtry
3   23
person madawa    schedule 28.01.2015

arrow_upward
0
arrow_downward

Метод "rf" в каретке использует пакет randomForest и функцию randomForest. Это если вы хотите запустить базовый случайный лес "вручную", без курсора.

person BBrill    schedule 27.01.2015
comment
Но дело в том, что я хочу использовать настроенные параметры из каретки. - person madawa; 28.01.2015