caret (version 5.07-001)

knnreg: k-Nearest Neighbour Regression

Description

$k$-nearest neighbour clasregressionsification that can return the average value for the neighbours.

Usage

## S3 method for class 'default':
knnreg(x, ...)

## S3 method for class 'formula': knnreg(formula, data, subset, na.action, k = 5, ...)

## S3 method for class 'matrix': knnreg(x, y, k = 5, ...)

## S3 method for class 'data.frame': knnreg(x, y, k = 5, ...)

knnregTrain(train, test, y, k = 5, use.all=TRUE)

Arguments

Value

Details

knnreg is similar to ipredknn and knnregTrain is a modification of knn. The underlying C code from the class pacakge has been modifed to return average outcome.

Examples

Run this code
data(BloodBrain)

inTrain <- createDataPartition(logBBB, p = .8)[[1]]

trainX <- bbbDescr[inTrain,]
trainY <- logBBB[inTrain]

testX <- bbbDescr[-inTrain,]
testY <- logBBB[-inTrain]

fit <- knnreg(trainX, trainY, k = 3)

plot(testY, predict(fit, testX))

Run the code above in your browser using DataCamp Workspace