Создание графика обучающих и проверочных наборов с использованием функции поезда кареты

Question

Я использую функцию поезда Карета в R, чтобы создать модель с использованием GBM. Я использовал повторную перекрестную проверку с 5 повторениями, то есть будет 50 образцов. Я хочу спросить, есть ли способ отобразить результаты по-другому, так что график показывает итерации повышения по оси x и auc по оси y, а внутри он показывает результаты, полученные при выборе лучшего параметра, но отдельная линия для тренировочных складок и тестовых складок. Это может быть получено, когда вы используете функцию "gbm" из пакета gbm и используете "gbm.perf" вместе с техникой выборки, чтобы построить кривую обучения и проверки для отклонения.

Можно ли сделать то же самое с функцией поезда Карета?

Благодарю.

Answer 1

В вашем объекте вставки, если вы использовали method='gbm', вы можете выбрать атрибут 'finalModel', который является результирующим объектом gbm. Например, если ваш железнодорожный объект называется "а", то

gbm_model <- $ finalModel

С помощью gbm_model вы можете запустить функции, встроенные в пакет gbm.