reg_knn: knn regression

View source: R/reg_knn.R

reg_knnR Documentation

knn regression

Description

Creates a regression object that uses the K-Nearest Neighbors (knn) method for regression

Usage

reg_knn(attribute, k)

Arguments

attribute

attribute target to model building

k

number of k neighbors

Value

returns a knn regression object

Examples

data(Boston)
model <- reg_knn("medv", k=3)

# preparing dataset for random sampling
sr <- sample_random()
sr <- train_test(sr, Boston)
train <- sr$train
test <- sr$test

model <- fit(model, train)

test_prediction <- predict(model, test)
test_predictand <- test[,"medv"]
test_eval <- evaluate(model, test_predictand, test_prediction)
test_eval$metrics

daltoolbox documentation built on Nov. 3, 2024, 9:06 a.m.