K近鄰法的R語言簡單實現(knn)
數據集采用R語言內置iris
查看數據集前6個觀測
head(iris)
k近鄰法
1.數據集分為訓練集和測試集
index <- sample(1:nrow(iris),100)
iris.train <- iris[index,]
iris.test <- iris[-index,]
train <- iris.train[,-5]
test <- iris.test[,-5]
2.加載K近鄰的class包
library(class)
3.利用KNN算法對測試集進行分類
result.KNN <- knn(train,test,cl=iris.train$Species)
4.生成結果集的混淆矩陣
table(result.KNN,iris.test$Species)
