【机器学习】：python：KNN简单的判别预测

版权声明：转载请联系博主。 https://blog.csdn.net/sunyaowu315/article/details/82907690

KNN简单的判别预测

# -*- coding: utf-8 -*-
"""
Created on Wed Aug  1 12:48:16 2018

@author: sunyaoowu315
"""

import numpy as np
from sklearn import neighbors
knn = neighbors.KNeighborsClassifier()   #取得 knn 分类器
data = np.array([[3,104],[2,100],[1,81],[101,10],[99,5],[98,2]])
labels = np.array([1,1,1,2,2,2])
knn.fit(data,labels)
knn.predict(np.array([18,90]).reshape(1,-1))