# -*- 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的数据集如下: