在R中查看kNN分类结果

时间:2014-08-05 15:23:25

标签: 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'。

感谢您的帮助!

1 个答案:

答案 0 :(得分:0)

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

只是不要使用摘要...它应该为每个测试观察输出预测类的向量。