1. 首页
  2. 人工智能
  3. 机器学习
  4. 使用Python实现决策树对莺尾花数据进行分类

使用Python实现决策树对莺尾花数据进行分类

上传者: 2023-06-28 10:32:51上传 PY文件 557B 热度 26次

首先,我们使用Python中的load_iris()函数加载莺尾花数据集。然后,我们将数据集分为特征(X)和标签(y)两部分进行存储。接下来,我们利用train_test_split()方法将数据集划分为训练集和测试集。创建一个决策树分类器对象,使用fit()方法对模型进行训练。使用score()方法测试模型性能并计算准确率。最后,我们输出模型的准确率。

下载地址
用户评论