1. Learn
    2. /
  2. 课程
    3. /
  3. Python 树模型机器学习
Connected

道练习

Entropy 与 Gini 指数对比

在本练习中，您将比较 dt_entropy 在测试集上的准确率与另一棵名为 dt_gini 的树的准确率。dt_gini 在相同的数据集上训练，参数也相同，唯一不同的是信息准则设置为了 Gini 指数（通过关键字 'gini'）。

X_test、y_test、dt_entropy，以及对应 dt_gini 在测试集上准确率的 accuracy_gini 已在您的工作区中提供。

说明

100 XP
  • 从 sklearn.metrics 导入 accuracy_score。
  • 预测 dt_entropy 的测试集标签，并将结果赋给 y_pred。
  • 评估 dt_entropy 在测试集上的准确率，并将结果赋给 accuracy_entropy。
  • 查看 accuracy_entropy 和 accuracy_gini。