Python编程实现用KNN算法对红酒分类功能

Python编程实现⽤KNN算法对红酒分类功能⼀、任务要求

导⼊红酒数据集（load_wine），编写Python代码，完成以下任务：

1、实现计算平均酒精含量的功能；

2、实现对数据的标准化；遐迩贯珍

3、使⽤kNN算法实现红酒分类功能

⼆、代码实现

from sklearn.datasets import load_wine

import pandas as pd

ighbors import KNeighborsClassifier

from sklearn.preprocessing import StandardScaler

from sklearn import preprocessing

del_selection import train_test_split

wine = load_wine()#导⼊数据集

谷泉会议中心data = wine['data']

巴厘路线图target = wine['target']

at([pd.DataFrame(data),pd.DataFrame(target)],axis=1)#⽤pandas合并数据集

# print(sample.head(3))

alcohol_mean = sample[10].mean()#酒精的指标在第11列，对酒精含量求均值

print("1、计算平均酒精含量:",alcohol_mean)

#数据集划分为训练集，测试集

data_train,data_test,target_train,target_test = train_test_split(data,target,test_size=0.2,random_state=125)

#实现对数据的标准化

print("2、实现对数据的标准化")

stdScaler = StandardScaler().fit(data_train)

data_std_train = ansform(data_train)

早熟禾属data_std_test = ansform(data_test)

# 使⽤KNN算法实现红酒的分类功能

print("3、使⽤KNN算法实现红酒的分类功能")

model=KNeighborsClassifier(n_neighbors=5)

model.fit(data_train,target_train)

pre=model.predict(data_test)

acc=model.score(data_test,target_test)

print("模型在测试集上的精度为:",acc)

北京青年工社

2012年湖北高考数学运⾏结果：

本文发布于:2024-09-22 03:37:36，感谢您对本站的认可！

标签：实现功能红酒分类算法数据酒精

留言与评论（共有 0 条评论）