diff --git a/车辆12分类数据集/第11组--张朋飞/案例代码/iris.py b/车辆12分类数据集/第11组--张朋飞/案例代码/iris.py new file mode 100644 index 0000000..8385f24 --- /dev/null +++ b/车辆12分类数据集/第11组--张朋飞/案例代码/iris.py @@ -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)