首页 > 解决方案 > 我们如何在 sklearn 中返回 K 最近邻?

问题描述

sklearn 中的 KNeighborsClassifier 预测了计算距离最短的 k 个实例的类,对吧?但是,我们可以返回他用来做这个预测的实例吗?

例如:如果 knn 由于与 Mary 和 Alexa 的距离而计算出 Ana 应该是女性,我们可以返回 ['Mary','Alexa']

标签: pythonmachine-learningscikit-learnknn

解决方案


有方法kneighbors,它返回每个点的邻居的索引和距离。

首先,您需要在训练数据集上拟合算法,然后您可以要求提供k neighbors一些特定的项目。


推荐阅读