python导入knn库_Python机器学习之scikit-learn库中KNN算法的封装与使用方法

本文实例讲述了Python机器学习之scikit-learn库中KNN算法的封装与使用方法。分享给大家供大家参考，具体如下：

1、工具准备，python环境，pycharm

2、在机器学习中，KNN是不需要训练过程的算法，也就是说，输入样例可以直接调用predict预测结果，训练数据集就是模型。当然这里必须将训练数据和训练标签进行拟合才能形成模型。

3、在pycharm中创建新的项目工程，并在项目下新建KNN.py文件。

import numpy as np

from math import sqrt

from collections import Counter

class KNNClassifier:

def __init__(self,k):

"""初始化KNN分类器"""

assert k >= 1

"""断言判断k的值是否合法"""

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[