西瓜决策树python代码
时间: 2024-05-30 09:06:04 浏览: 126
西瓜书《机器学习》---第四章 决策树python代码实现
5星 · 资源好评率100%
西瓜书中提到了C4.5算法和CART算法,它们都是决策树的经典算法。如果你想了解决策树算法的python实现,可以使用scikit-learn库中的DecisionTreeClassifier类。下面是一个简单的例子:
```python
from sklearn.tree import DecisionTreeClassifier
from sklearn.datasets import load_iris
# 加载数据集
iris = load_iris()
# 创建决策树分类器对象
clf = DecisionTreeClassifier()
# 训练模型
clf.fit(iris.data, iris.target)
# 预测新样本
print(clf.predict([[5.1, 3.5, 1.4, 0.2]]))
```
以上代码使用iris数据集进行训练,然后预测新样本。你可以根据需要修改数据集和决策树分类器的参数。
阅读全文