机器学习(二)使用sklearn库的datasets练习KNN分类

来源:互联网 发布:网络用语lq什么意思 编辑:程序博客网 时间:2024/06/06 05:11
# -*- coding: utf-8 -*-"""Created on Tue Oct 25 20:10:51 2016@author: SIriustest sklearn"""import numpy as npfrom sklearn import datasetsfrom sklearn.cross_validation import train_test_splitfrom sklearn.neighbors import KNeighborsClassifieriris=datasets.load_iris()data_x=iris.datadata_y=iris.target#print 'X :',data_x[0:10,:] #四个属性,花瓣的长宽,叶的长宽#print 'Y:',data_y[0:10]  #0代表青风藤,1代表云腾,2代表紫藤x_train,x_test,y_train,y_test=train_test_split(                              data_x,data_y,test_size=0.3) #把数据打乱knn=KNeighborsClassifier() knn.fit(x_train,y_train) # 训练knn模型print knn.predict(x_test)print y_test

运行结果如下:
这里写图片描述

这里iris的数据集如下:
这里写图片描述

0 0
原创粉丝点击