（KNN）iris种类预测

最新推荐文章于 2023-04-12 01:03:51 发布

付修磊

最新推荐文章于 2023-04-12 01:03:51 发布

阅读量753

点赞数

分类专栏：通往Kaggle之路

本文链接：https://blog.csdn.net/qq_38669138/article/details/79337590

版权

通往Kaggle之路专栏收录该内容

6 篇文章 0 订阅

订阅专栏

# -*- coding: cp936 -*-
from sklearn.datasets import load_iris
iris=load_iris()
iris.data.shape
print iris.DESCR

from sklearn.cross_validation import train_test_split
X_train,X_test,y_train,y_test=train_test_split(iris.data,iris.target,test_size=0.25,random_state=33)

from sklearn.preprocessing import StandardScaler
from sklearn.neighbors import KNeighborsClassifier


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 KNN is',knc.score(X_test,y_test)

from sklearn.metrics import classification_report
print classification_report(y_test,y_predict,target_names=iris.target_names)

import sklearn
model=sklearn.neighbors.KNeighborsClassifier(n_neighbors=5, weights="uniform", algorithm="auto", leaf_size=30, p=2, metric="minkowski", metric_params=None, n_jobs=1)

或者这样子

# -*- coding: cp936 -*-
from sklearn.datasets import load_iris
iris=load_iris()
iris.data.shape
print iris.DESCR

from sklearn.cross_validation import train_test_split
X_train,X_test,y_train,y_test=train_test_split(iris.data,iris.target,test_size=0.25,random_state=33)

from sklearn.preprocessing import StandardScaler
from sklearn.neighbors import KNeighborsClassifier


ss=StandardScaler()
X_train=ss.fit_transform(X_train)
X_test=ss.transform(X_test)





import sklearn
model=sklearn.neighbors.KNeighborsClassifier(n_neighbors=5, weights="uniform", algorithm="auto", leaf_size=30, p=2, metric="minkowski", metric_params=None, n_jobs=1)

model.fit(X_train,y_train)
y_predict=model.predict(X_test)

print 'The accuracy of KNN is',model.score(X_test,y_test)

from sklearn.metrics import classification_report
print classification_report(y_test,y_predict,target_names=iris.target_names)