机器学习（二）使用sklearn库的datasets练习KNN分类

# -*- coding: utf-8 -*-
"""
Created on Tue Oct 25 20:10:51 2016

@author: SIrius
test sklearn
"""

import numpy as np
from sklearn import datasets
from sklearn.cross_validation import train_test_split
from sklearn.neighbors import KNeighborsClassifier

iris=datasets.load_iris()
data_x=iris.data
data_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的数据集如下：