#KNN最近邻分类算法 from sklearn.neighbors import KNeighborsClassifier from sklearn.datasets import load_digits from sklearn.model_selection import train_test_split x=wine_data.iloc[:,1:].values y=wine_data.iloc[:,0].values x_train,x_test,y_train,y_test=train_test_split(x,y,test_size=0.3,random_state=125) dtr=KNeighborsClassifier() dtr.fit(x_train,y_train) dtr.score(x_test,y_test)
时间: 2023-07-28 22:00:24 浏览: 104
K最近邻算法(KNN)---sklearn+python实现方式
这段代码使用sklearn库中的K最近邻(K Nearest Neighbor,KNN)分类器对wine_data数据集进行分类建模。具体来说,代码首先从wine_data中分离出特征数据x和目标数据y,其中x为除第一列以外的所有列,y为第一列的数据。然后,使用train_test_split函数将x和y分别划分为训练集和测试集,其中test_size=0.3表示将30%的数据作为测试集,random_state=125表示随机种子,以确保每次划分结果相同。
接着,使用KNeighborsClassifier函数定义一个KNN分类器模型。然后,使用fit函数对训练集进行拟合,得到一个训练好的模型。最后,使用score函数计算模型在测试集上的精度,并将结果输出。
需要注意的是,KNN算法通过计算输入样本与训练集中所有样本之间的距离来确定输入样本的类别。因此,在实际应用中,需要根据数据集的特点选择合适的距离度量方式和K值,并对模型进行评估和调整,以提高模型的性能。
阅读全文