扫码加入数据分析学习群

import numpy as np
2. from sklearn import neighbors
3. knn = neighbors.KNeighborsClassifier() #取得 knn 分类器
4. data = np.array([[3,104],[2,100],[1,81],[101,10],[99,5],[98,2]]) # <span style="font-family:Arial, Helvetica, sans-serif;">data 对应着
打斗次数和接吻次数</span>
5. labels = np.array([1,1,1,2,2,2]) #<span style="font-family:Arial, Helvetica, sans-serif;">labels 则是对应 Romance 和 Action</span>
6. knn.fit(data,labels) #导入数据进行训练'''
7. #Out： KNeighborsClassifier(algorithm='auto', leaf_size=30, metric='minkowski',
8. metric_params=None, n_jobs=1, n_neighbors=5, p=2,
9. weights='uniform')
10. knn.predict([18,90])