k近邻算法(KNN)

发表于 | 分类于
字数统计: 998字 | 阅读时长 ≈ 4 分钟

k近邻算法

k近邻算法( k-Nearest Neighbor)是一种监督学习
优点:精度高、对异常值不敏感、无数据输入假定
缺点:计算复杂度高、空间复杂度高

工作原理

给定测试样本, 基于某种距离度量找出训练集中与其最靠近的 个训练样本, 然后基于这 个邻居的信息来进行预测

一般流程包括：

  1. 计算已知类别数据集中的点与当前点之间的距离
  2. 按照距离递增次序排序
  3. 选取与当前点距离最小的k个点
  4. 确定前k个点所在类别的出现频率
  5. 返回前k个点出现频率最高的类别作为当前点的预测分类

分类任务中可选择这 个样本中出现最多的类别标记作为预测结果

回归任务中可将这 个样本的实值输出标记的平均值作为预测结果

k值的选择

当 取不同值时, 分类结果会有显著不同

K值过小：特征空间被划分为更多子空间，整体模型变复杂，预测结果会对近邻点十分敏感，预测就会出错容易发生过拟合
K值过大：近邻误差会偏大，距离较远的点也会同样对预测结果产生影响，使得预测结果产生较大偏差，此时模型容易发生欠拟合

距离度量

若采用不同的距离计算方式也会导致分类结果有显著不同

对函数 ，若它是一个距离度量，则需满足一些基本性质

给定样本 与
最常用的是闵可夫斯基距离(Minkowski distance）

当 时, 闵可夫斯基距离即欧氏距离 (Euclidean distance)

欧几里得空间中两点间直线距离、真实距离或者向量的自然长度

当 时, 闵可夫斯基距离即曼哈顿距离(Manhattan distance)

在欧几里德空间的固定直角坐标系上两点所形成的线段对轴产生的投影的距离总和，也称街区距离

代码实现

FROM IMOOC机器学习

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
import numpy as np
from math import sqrt
from collections import Counter

class KNNClassifier:

    def __init__(self, k):
        """初始化kNN分类器"""
        assert k >= 1, "k must be valid"
        self.k = k
        self._X_train = None
        self._y_train = None

    def fit(self, X_train, y_train):
        """根据训练数据集X_train和y_train训练kNN分类器"""
        assert X_train.shape[0] == y_train.shape[0], \
            "the size of X_train must be equal to the size of y_train"
        assert self.k <= X_train.shape[0], \
            "the size of X_train must be at least k."

        self._X_train = X_train
        self._y_train = y_train
        return self

    def predict(self, X_predict):
        """给定待预测数据集X_predict，返回表示X_predict的结果向量"""
        assert self._X_train is not None and self._y_train is not None, \
                "must fit before predict!"
        assert X_predict.shape[1] == self._X_train.shape[1], \
                "the feature number of X_predict must be equal to X_train"

        y_predict = [self._predict(x) for x in X_predict]
        return np.array(y_predict)

    def _predict(self, x):
        """给定单个待预测数据x，返回x的预测结果值"""
        assert x.shape[0] == self._X_train.shape[1], \
            "the feature number of x must be equal to X_train"

        distances = [sqrt(np.sum((x_train - x) ** 2))
                     for x_train in self._X_train]
        nearest = np.argsort(distances)

        topK_y = [self._y_train[i] for i in nearest[:self.k]]
        votes = Counter(topK_y)

        return votes.most_common(1)[0][0]

    def score(self, X_test, y_test):
        """根据测试数据集 X_test 和 y_test 确定当前模型的准确度"""

        y_predict = self.predict(X_test)
        return self.accuracy_score(y_test, y_predict)

    def __repr__(self):
        return "KNN(k=%d)" % self.k

    def accuracy_score(y_true, y_predict):
        """计算y_true和y_predict之间的准确率"""
        assert len(y_true) == len(y_predict), \
            "the size of y_true must be equal to the size of y_predict"
    
        return np.sum(y_true == y_predict) / len(y_true)

参考

LP距离、欧式距离、曼哈顿距离、切比雪夫距离

什么是KNN算法？

IMOOC机器学习

机器学习-周志华

Machine Learning in Action by Peter Harrington