开始使用免费开始使用

受限与不受限的决策树

在本练习中,我们将回到上一章节使用的宝可梦数据集。回顾一下,我们的目标是预测某只宝可梦是否为传说宝可梦。

这里,您将构建两个独立的决策树分类器。第一个模型中,您只指定参数 min_samples_leafmin_samples_split,不设置最大深度,这样树可以在没有限制的情况下充分生长。

第二个模型中,您将通过限制决策树的深度来施加一些约束。随后对比两个模型,您将更好地理解"弱"学习器的概念。

本练习是课程的一部分

Python 中的集成方法

查看课程

交互式实操练习

通过完成这段示例代码来试试这个练习。

# Build unrestricted decision tree
clf = ____
clf.fit(X_train, y_train)

# Predict the labels
pred = clf.predict(X_test)

# Print the confusion matrix
cm = confusion_matrix(y_test, pred)
print('Confusion matrix:\n', cm)

# Print the F1 score
score = f1_score(y_test, pred)
print('F1-Score: {:.3f}'.format(score))
编辑并运行代码