查看R中的kNN分类结果



我想看看我的训练数据是如何分类的。我可以让算法运行得很好,并且可以看到我的训练数据中有多少观察值被分类为"A"或"B"(我的两个级别)。但我想看看训练数据中每个单独的观察结果是如何分类的。如果我能把分类结果作为一个新列附加到训练数据上,那就太好了。

到目前为止我写的是:

head(train,n=5) 
 x  y
1 30 16
2 15  5
3 25 27
4 15 30
5  5 22
head(test,n=5)
  x  y
40 79 80
41 60 78
42 70 53
43 38 72
44  2 53
 cl
 [1] A A A A A A A A A A A A A A A A A A A B B B B B B B B B B B B B B B B B B B
[39] B
Levels: A B

 library(class)
 results = summary(knn(train,test,cl,k=3))
 results
 A B 
 4 6 

我正试图查看测试数据中的哪些观察结果属于这4个A和6个B。

谢谢你的帮助!

predictions = knn(train,test,cl,k=3)

只是不要用总结…它应该为每个测试观察输出一个预测类的向量。

最新更新