Как запустить регрессию вместо классификации в случайном лесу в R? - PullRequest
0 голосов
/ 15 января 2019

Я запускаю этот код в R

model_bf_df <- randomForest(Amountspent.x ~ ., data = train_bf_df, ntree = 130, mtry=59, importance = TRUE, do.trace=TRUE)

Amountspent.x - это переменная продолжения, которая принимает значения из набора данных поезда, который я ранее отделил от основного набора данных. Я получаю тип классификации вместо регрессии. Какие изменения я должен сделать? или я что-то упустил?

1 Ответ

0 голосов
/ 17 января 2019

Пакет randomForest автоматически определяет тип проблемы на основе класса y, Amountspent.x в вашем случае. Ваш DV должен быть фактором или чем-то отличным от числового. Убедитесь, что вы правильно преобразовали его в числовой формат. Проверьте с class(train_bf_df$Amountspent.x).

...