决策树模型构建与预测:最大深度5,最小叶子节点数10
构建决策树模型,设置最大深度为5,最小叶子节点数为10
首先,我们需要导入必要的库并创建决策树模型,设置最大深度为5,最小叶子节点数为10:
from sklearn.tree import DecisionTreeClassifier
# 创建决策树模型
clf = DecisionTreeClassifier(max_depth=5, min_samples_leaf=10)
接下来,我们可以使用训练数据训练模型:
# 训练模型
clf.fit(X_train, Y_train)
训练完成后,可以使用测试数据评估模型性能:
# 在测试集上评估模型性能
score = clf.score(X_test, Y_test)
如何预测数据内容
使用训练好的决策树模型进行数据预测非常简单,例如:
# 假设有一组新数据X_new需要进行预测
Y_pred = clf.predict(X_new)
# Y_pred为预测结果,可以与真实结果进行比较,评估模型性能
通过以上步骤,我们可以构建一个决策树模型,并使用它进行数据预测。
原文地址: https://www.cveoy.top/t/topic/f1fV 著作权归作者所有。请勿转载和采集!