决策树算法python代码
时间: 2023-06-08 11:06:17 浏览: 359
以下是一个简单的决策树算法Python代码示例:
```python
from sklearn.tree import DecisionTreeClassifier
from sklearn.datasets import load_iris
# 载入Iris数据集
iris = load_iris()
X = iris.data
y = iris.target
# 训练模型
clf = DecisionTreeClassifier()
clf.fit(X, y)
# 预测
new_data = [[5.2, 3.1, 1.6, 0.4]]
print(clf.predict(new_data))
```
这个示例使用scikit-learn库中的`DecisionTreeClassifier`类实现了决策树算法。首先载入了Iris数据集,然后训练了一个决策树分类器。最后,使用这个分类器对新的数据进行预测。
阅读全文
相关推荐















