предсказать цель новых данных с использованием Nnet R

Question

Я сталкиваюсь с проблемами предсказать функцию в пакете Nnet R.

Я воспитал Nnet модель в R с данными поезда, но когда я пытаюсь предсказать для новых испытательных образцов I я получаю больше результатов, чем количество тестовых образцов.

Данные моего поезда содержат 3673 строки, и мой тест имеет 1225. Но когда я прогнозирую данные теста, я получаю 3673 целевых выхода. Я вставил свой код ниже.

index<-1:nrow(wine_data) 
set.seed(1) 
train_ind <- sample(index,.75*nrow(wine_data)) 
train<-wine_data[train_ind,] 
test<-wine_data[-train_ind,] 

model.nnet<-nnet(formula=f,data=train,linout=FALSE,size=10) 

predicted<-predict(model.nnet,data=test,type='class') 

Выше скриншот г консоли. Я делаю что-то неправильно здесь. Пожалуйста, помогите мне. Спасибо.

Answer 1

В последней строке кода у вас есть

данные = поезд

Так что либо положить данных = тест, или new.data = тест (в зависимости от того, что функция предсказать для Nnet пакет говорит)

Функция прогнозирования не нуждается в данных поезда (большую часть времени), так как она просто запоминает спецификацию модели и может применяться к новым данным с одинаковыми функциями. или использовать

predicted<-predict(model.nnet, test,type="class") 

Answer 2

У меня проблема. Когда я заменяю данные = тест с тестом в функции прогнозирования он работает. Это мой модифицированный код прогноза.

< предсказал -predict (model.nnet, тест, тип = 'класс')