Прогнозы KNN возвращают меньше строк, чем строки обучающих данных - PullRequest
Новая
       28

Прогнозы KNN возвращают меньше строк, чем строки обучающих данных

0 голосов
/ 03 мая 2019

Я создаю прогнозы из обучающего набора данных, который имеет 207 строк, но возвращаемые прогнозы состоят из 90 строк. Я не могу понять, в чем проблема. Может кто-нибудь указать, где я иду не так? 

# Fit the model 
knn_optimal <-  kknn(HeartDisease~., 
                 train = heart_training, 
                 test = heart_test,
                 k = 33, distance = 2)
knn_optimal

#For Class:
knn_pred_0.5=predict(knn_optimal, heart_training, type="raw", cutoff=0.5)
knn_pred_0.5

View(as.data.frame(knn_pred_0.5))

#For PP:
knn_predpp=predict(knn_optimal, heart_training, type="prob", cutoff=0.5)
knn_predpp


View(as.data.frame(knn_predpp))

# Create results data frame with training data
knn_training_results <- data.frame(heart_training, knn_pred_0.5, knn_predpp)

# Check results
knn_training_results %>% slice(1:5)
...