my_knn_cv: k-nearest-neighbor function

In RuofengT/stat302package: Project 3 Stat 302

Description

This function performs k nearest neighbors with k-fold cross validation.

Usage

my_knn_cv(train, cl, k_nn, k_cv)

Arguments

train	input data frame
cl	true class value of your training data
k_nn	integer representing the number of neighbors
k_cv	integer representing the number of folds

Value

a list with objects "class": a vector of the predicted class for all observations; "cv_err": a numeric with the cross-validation misclassification error

Examples

dat <- na.omit(my_penguins[, c("species", "bill_length_mm", "bill_depth_mm",
                               "flipper_length_mm", "body_mass_g")])
cl <- dat$species
dat <- dat[, c("bill_length_mm", "bill_depth_mm",
               "flipper_length_mm", "body_mass_g")]
my_knn_cv(dat, cl, 1, 5)

RuofengT/stat302package documentation built on Dec. 18, 2021, 11:54 a.m.