KNN算法实例(酒的分类)【机器学习算法一KNN4】

1.代码

1. 未画图

from sklearn.datasets import load_wine
wine=load_wine()#Bunch对象，包含键和数值

print("=========================================")
print("红酒数据集中的键：\n{}".format(wine.keys()))
print("数据概况：{}".format(wine['data'].shape))
print(wine['DESCR'])#数据集描述
print("=========================================\n")

from sklearn.model_selection import train_test_split
#默认情况下，75%数据作为训练集，25%数据作为测试集，random_state指定伪随机数供数据集拆分(=0每次伪随机数不同)
X_train,X_test,y_train,y_test=train_test_split(wine['data'],wine['target'],random_state=0)

print("=========================================")
print('Shape\nX_train:{}, X_test:{}, y_train:{}, y_test:{}\n'.format(X