目录
什么是k近邻算法
模型的三个基本要素
构造kd树
搜索kd树
Python代码(sklearn库)
什么K近邻算法(k-Nearest Neighbor,kNN) |
引例
假设有数据集,其中前6部是训练集(有属性值和标记),我们根据训练集训练一个KNN模型,预测最后一部影片的电影类型。
首先,将训练集中的所有样例画入坐标系,也将待测样例画入
然后计算待测分类的电影与所有已知分类的电影的欧式距离
接着,将这些电影按照距离升序排序,取前k个电影,假设k=3,那么我们得到的电影依次是《He‘s Not Really Into Dudes》、《Beautiful Woman》和《California Man》。而这三部电影全是爱情片,因此我们判定未知电影是爱情片。
时间: 2025-01-07 18:00:26