平时作业

Signed-off-by: 张朋飞 <13190662+z1479344137@user.noreply.gitee.com>
2023-07-16 02:50:42 +00:00
parent 3ceac6f938
commit 63fdac5217
1 changed files with 25 additions and 0 deletions
@@ -0,0 +1,25 @@
+from sklearn.datasets import load_iris
+from sklearn.model_selection import train_test_split
+from sklearn.neighbors import KNeighborsClassifier
+from sklearn.metrics import accuracy_score
+
+# 加载Iris数据集
+iris = load_iris()
+X = iris.data
+y = iris.target
+
+# 将数据集拆分为训练集和测试集
+X_train, X_test, y_train, y_test = train_test_split(X, y, test_size=0.2, random_state=42)
+
+# 分类器
+knn = KNeighborsClassifier(n_neighbors=3)
+
+# 在训练集上训练模型
+knn.fit(X_train, y_train)
+
+# 在测试集上进行预测
+y_pred = knn.predict(X_test)
+
+# 计算准确率
+accuracy = accuracy_score(y_test, y_pred)
+print('准确率：', accuracy)