构建决策树模型并进行预测

本教程将演示如何构建一个决策树模型,并使用它对新数据进行预测。我们将设置最大深度为5,最小叶子节点数为10。

1. 构建模型

首先,使用 DecisionTreeClassifier 类构建一个决策树模型,并设置 max_depthmin_samples_leaf 参数:

clf = DecisionTreeClassifier(max_depth=5, min_samples_leaf=10)

然后,使用训练数据 X_trainY_train 对模型进行训练:

clf.fit(X_train, Y_train)

最后,使用测试数据 X_testY_test 评估模型性能:

score = clf.score(X_test, Y_test)

2. 预测新数据

假设我们有一个新的数据样本,其特征值为 [5.1, 3.5, 1.4, 0.2],我们可以使用训练好的决策树模型进行预测:

# 预测新样本
new_sample = [[5.1, 3.5, 1.4, 0.2]]
predicted_class = clf.predict(new_sample)

# 输出预测结果
print('Predicted class:', predicted_class)

本教程使用Python代码示例,详细讲解了如何构建一个决策树模型,设置最大深度和最小叶子节点数,并使用它进行数据预测。希望本教程能够帮助您更好地理解决策树模型的构建与应用。

决策树模型构建与预测:最大深度5,最小叶子节点10

原文地址: https://www.cveoy.top/t/topic/f1fW 著作权归作者所有。请勿转载和采集!

免费AI点我,无需注册和登录