当前位置:   article > 正文

python-KNN分类(1):调用KNeighborsClassifier()实现_knn分类调用

knn分类调用

KNeighborsClassifier(): 

  1. '''
  2.  KNeighborsClassifier(n_neighbors=5, weights='uniform', 
  3.                       algorithm='auto', leaf_size=30, 
  4.                       p=2, metric='minkowski', 
  5.                       metric_params=None, n_jobs=1, **kwargs)
  6.  n_neighbors: 默认值为5,表示查询k个最近邻的数目
  7.  algorithm:   {‘auto’, ‘ball_tree’, ‘kd_tree’, ‘brute’},指定用于计算最近邻的算法,auto表示试图采用最适合的算法计算最近邻
  8.  leaf_size:   传递给‘ball_tree’或‘kd_tree’的叶子大小
  9.  metric:      用于树的距离度量。默认'minkowski与P = 2(即欧氏度量)
  10.  n_jobs:      并行工作的数量,如果设为-1,则作业的数量被设置为CPU内核的数量
  11.  查看官方api:http://scikit-learn.org/dev/modules/generated/sklearn.neighbors.KNeighborsClassifier.html#sklearn.neighbors.KNeighborsClassifier
  12. '''

示例:

  1. # -*- coding: utf-8 -*-
  2. """
  3. Created on Fri Feb 15 18:01:45 2019
  4. @author: Administrator
  5. """
  6. import numpy as np
  7. from sklearn import neighbors
  8. import warnings
  9. warnings.filterwarnings('ignore')  # warning信息不打印,可有可无
  10. knn = neighbors.KNeighborsClassifier()  # 取得knn分类器
  11. data = np.array([[1, 1, 1, 1],
  12.               [0.5, 1, 1, 1],
  13.               [0.1, 0.1, 0.1, 0.1],
  14.               [0.5, 0.5, 0.5, 0.5],
  15.               [1, 0.8, 0.3, 1],
  16.               [0.6, 0.5, 0.7, 0.5],
  17.               [1, 1, 0.9, 0.5],
  18.               [1, 0.6, 0.5, 0.8],
  19.               [0.5, 0.5, 1, 1],
  20.               [0.9, 1, 1, 1],
  21.               [0.6, 0.6, 1, 0.1],
  22.               [1, 0.8, 0.5, 0.5],
  23.               [1, 0.1, 0.1, 1],
  24.               [1, 1, 0.7, 0.3],
  25.               [0.2, 0.3, 0.4, 0.5],
  26.               [0.5, 1, 0.6, 0.6]
  27.               ])
  28. labels = np.array(['美女',
  29.           '淑女',
  30.           '丑女',
  31.           '一般型',
  32.           '淑女',
  33.           '一般型',
  34.           '美女',
  35.           '一般型',
  36.           '淑女',
  37.           '美女',
  38.           '丑女',
  39.           '可爱型',
  40.           '可爱型',
  41.           '淑女',
  42.           '丑女',
  43.           '可爱型'
  44.           ])
  45. knn.fit(data, labels)  # 导入数据进行训练
  46. print('预测类型为:', knn.predict([[0.8, 1, 1, 1]]))

结果:

声明:本文内容由网友自发贡献,转载请注明出处:【wpsshop】
推荐阅读
相关标签
  

闽ICP备14008679号