我想看看我的训练数据是如何分类的。我可以让算法运行得很好,并且可以看到我的训练数据中有多少观察值被分类为"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)
只是不要用总结…它应该为每个测试观察输出一个预测类的向量。