R - avNNet - функция Predict возвращает меньше строк, чем в тестовой функции

Я пробую avNNet для одного из проектов классификации с 1 и 0.

В моем тестовом наборе данных у меня 116293. Но вывод функции прогнозирования возвращает только 106388 записей.

И шокер - все 0!!! Я могу сделать какую-то ошибку. Пожалуйста, помогите мне использовать avNNet.

Код выглядит следующим образом:

nnet_train = avNNet(WnvPresent~. ,data = xtrain, size=10)
str(xtest) #116293 obs. of  16 variables
PredTest = predict(nnet_train, newdata = xtest)
str(PredTest) #Named num [1:106388] 
summary(PredTest)

r predict
person Vimal Kumar M    schedule 08.06.2015    source источник

Ответы (1)


arrow_upward
1
arrow_downward

Это происходит, когда в ваших данных есть столбцы с NA (xtest здесь). Predict() пропустит строки, в любом из столбцов которых есть NA.

Я знаю, что уже поздно, но я надеюсь, что это будет полезно для потомков.

person Abhimanu Kumar    schedule 25.05.2016