写一个决策树预测代码
时间: 2023-02-08 07:06:51 浏览: 122
决策树是一种常用的分类和回归算法。它能够帮助我们根据数据特征进行分类和预测。在 Python 中,可以使用 scikit-learn 库来训练和使用决策树模型。
以下是一个简单的决策树预测代码示例,假设你已经准备好了训练数据和测试数据。
``` python
from sklearn.tree import DecisionTreeClassifier
from sklearn.metrics import accuracy_score
# 加载数据
X_train, y_train = # 训练数据
X_test, y_test = # 测试数据
# 创建决策树模型
clf = DecisionTreeClassifier()
# 训练模型
clf.fit(X_train, y_train)
# 预测
y_pred = clf.predict(X_test)
# 评估模型准确率
acc = accuracy_score(y_test, y_pred)
print("Accuracy:", acc)
```
在这段代码中,我们首先导入了 DecisionTreeClassifier 和 accuracy_score,它们分别用于构建决策树模型和评估模型准确率。然后我们创建了一个 DecisionTreeClassifier 对象,并使用 fit 方法训练模型。最后使用 predict 方法对测试数据进行预测,并使用 accuracy_score 计算预测准确率。
至于决策树的参数调优或树的可视化,就需要额外的代码来进行操作.
另外这只是简单的预测样例,需要根据你具体数据
阅读全文