开始使用免费开始使用

决策树 vs. 神经网络

构建一个决策树分类器,根据包括年龄、受教育程度、每周工作时长等多种特征来预测收入水平,并提取模型学到的用于解释决策的规则。然后,将其与在相同数据上训练的 MLPClassifier 的性能进行比较。

X_trainX_testy_trainy_test 已为您预加载。accuracy_scoreexport_text 函数也已为您导入。

本练习是课程的一部分

Python 可解释性 AI

查看课程

交互式实操练习

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

model = DecisionTreeClassifier(random_state=42, max_depth=2)
model.fit(X_train, y_train)

# Extract the rules
rules = ____
print(rules)

y_pred = model.predict(X_test)

# Compute accuracy
accuracy = ____
print(f"Accuracy: {accuracy:.2f}")
编辑并运行代码