基于CART算法生成决策树 python编写
时间: 2024-05-04 17:18:37 浏览: 152
python 实现决策树算法(CART)
CART(Classification and Regression Trees)算法是一种用于决策树分类和回归的非常流行的算法。下面是一个基于Python实现的CART算法生成决策树的代码示例:
```python
from sklearn.tree import DecisionTreeClassifier
from sklearn.datasets import load_iris
from sklearn.model_selection import train_test_split
from sklearn.metrics import accuracy_score
iris = load_iris()
X_train, X_test, y_train, y_test = train_test_split(iris.data, iris.target, random_state=0)
clf = DecisionTreeClassifier(random_state=0)
clf.fit(X_train, y_train)
y_pred = clf.predict(X_test)
print("Accuracy:", accuracy_score(y_test, y_pred))
# Output:
# Accuracy: 0.9736842105263158
```
在这个示例中,我们使用了sklearn(scikit-learn)库中的DecisionTreeClassifier类来实现CART算法生成决策树。首先,我们加载了鸢尾花数据集并将其分成训练集和测试集。然后,我们创建了一个决策树分类器对象并使用训练集训练它。最后,我们用测试集测试了决策树的准确率。
需要注意的是,这只是一个简单的示例,实际上在使用CART算法生成决策树时,我们需要仔细选择参数以避免过拟合或欠拟合等问题。
阅读全文