Python实现机器算法-02-KNN

# -*- coding: UTF-8 -*-
"""

~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
@File         :   knn.py    
@Contact      :   ffzzyy@126.com
@License      :   (C)Copyright 2017-2019
@Author       :   ffzzyy
@Version      :   0.1
@Modify Time  :   2019/3/13 22:51
@Desciption

"""

import numpy as np
from sklearn import datasets
from collections import Counter

from sklearn.neighbors import KNeighborsClassifier


class KNNClassifier:
    def __init__(self, k):
        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):
        '''
        训练函数
        :param x_train:
        :param y_train:
        :return:
        '''
        self._x_train = x_train
        self._y_train = y_train
        return self

    def _predict(self, x):
        '