用KNN算法对鸢尾花数据进行分类(代码)

过程分为五步:

1. 获取数据
2. 划分数据集
3. 特征工程:标准化
4. KNN算法估计器
5. 模型评估(对比真实值和估计值 ; 计算准确率)

#coding=UTF-8


from sklearn.datasets import  load_iris
from sklearn.model_selection  import  train_test_split
from sklearn.preprocessing import StandardScaler
from sklearn.neighbors import  KNeighborsClassifier


def knn_selector():
    '''
    用KNN对数据进行分类
    '''
    #(1) 获取数据
    iris = load_iris()
    #(2) 划分数据集
    x_train,x_test,y_train,y_test = train_test_split(iris.data , iris.target , random_state= 6)
    #(3) 特征工程:标准化
    transfer  =  StandardScaler()
    x_train = transfer.fit_transform(x_train)
    x_test= transfer.transform(x_test)
    #(4) KNN算法预估器
    estimator =  KNeighborsClassifier(n_neighbors= 3)
    estimator.fit(x_train, y_train)
    #(5) 模型评估
    score = estimator.score(x_test,y_test)
    print("准确率为:\n", score)


if __name__=="__main__":
    knn_selector()
发布了35 篇原创文章 · 获赞 5 · 访问量 2422

猜你喜欢

转载自blog.csdn.net/qq_24884193/article/details/104089857
今日推荐