学习笔记:K近邻预测Iris(鸢尾)分类

代码如下:

# -*- coding: utf-8 -*-
"""
Created on Fri May 11 10:26:58 2018

@author: eagle
"""

# =============================================================================
# 用K近邻预测Iris(鸢尾)数据集
# =============================================================================


from sklearn.datasets import load_iris
from sklearn.cross_validation import train_test_split
from sklearn.preprocessing import StandardScaler
from sklearn.neighbors import KNeighborsClassifier
from sklearn.metrics import classification_report

#导入数据集
iris = load_iris()
#print(iris.data.shape)  #(150,4)共150条数据,4个属性
#查看数据描述
#print(iris.DESCR)

#对数据集进行分割(训练数据、测试数据)
X_train,X_test,y_train,y_test = train_test_split(iris.data,iris.target,test_size=0.25,random_state=33)

#数据标准化
ss = StandardScaler()
X_train = ss.fit_transform(X_train)
X_test = ss.transform(X_test)

#预测
knc=KNeighborsClassifier()
knc.fit(X_train,y_train)    #训练
y_predict = knc.predict(X_test)     #预测

#评价
print('The Accuracy of K-Nearest Neighbor Classifier is:',knc.score(X_test,y_test))
print(classification_report(y_test,y_predict,target_names=iris.target_names))

猜你喜欢

转载自blog.csdn.net/HorseRunningNoStop/article/details/80279558