python实现鸢尾花分类
时间: 2023-12-04 21:02:40 浏览: 17
基于Python BP神经网络实现鸢尾花的分类【100012157】
5星 · 资源好评率100%
可以使用scikit-learn库中的决策树模型来实现鸢尾花分类。具体实现代码可以参考以下示例:
```
from sklearn.datasets import load_iris
from sklearn.tree import DecisionTreeClassifier
from sklearn.model_selection import train_test_split
# 加载数据集
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=0)
# 构建决策树模型
clf = DecisionTreeClassifier()
clf.fit(X_train, y_train)
# 在测试集上进行预测
y_pred = clf.predict(X_test)
# 输出预测结果
print("预测结果:", y_pred)
```
当然,如果您有更多的问题,我也很乐意为您解答。
阅读全文