from sklearn import datasets from sklearn.neighbors import KNeighborsClassifier from sklearn.model_selection import train_test_split import matplotlib.pyplot as plt import numpy as np # 加载数据集 iris = datasets.load_iris() X = iris.data y = iris.target # 划分训练集和测试集 X_train, X_test, y_train, y_test = train_test_split(X, y, test_size=0.3) # 创建KNN分类器,K=3 knn = KNeighborsClassifier(n_neighbors=3) # 训练模型 knn.fit(X_train, y_train) # 在测试集上进行预测 y_pred = knn.predict(X_test) # 计算准确率 accuracy = knn.score(X_test, y_test) print("准确率为:", accuracy) # 绘制分类分布结果的二维图 colors = ['red', 'green', 'blue'] markers = ['o', 's', 'x'] for i, color, marker in zip(np.unique(y_pred), colors, markers): plt.scatter(X_test[y_pred == i, 0], X_test[y_pred == i, 1], color=color, marker=marker, label=iris.target_names[i]) plt.legend() plt.xlabel(iris.feature_names[0]) plt.ylabel(iris.feature_names[1]) plt.show()
时间: 2024-03-18 20:44:27 浏览: 116
机器学习-基于Python实现的机器学习算法之KNN.zip
这段代码使用了scikit-learn库中的K近邻算法实现对鸢尾花数据集的分类。首先,通过导入数据集并将其分为训练集和测试集。然后创建了一个KNeighborsClassifier对象,其中传递参数n_neighbors=3表示使用3个邻居来进行分类。接下来,使用fit()方法训练模型,使用predict()方法在测试集上进行预测。最后,使用score()方法计算模型在测试集上的准确率,并使用matplotlib库绘制了分类结果的二维图。
阅读全文