给我一段决策树的python代码
时间: 2023-09-07 18:04:10 浏览: 91
决策树python代码
### 回答1:
from sklearn import tree#[features]
features = [[140, "smooth"], [130, "smooth"], [150, "bumpy"], [170, "bumpy"]]
#[labels]
labels = ["apple", "apple", "orange", "orange"]#[classifier]
clf = tree.DecisionTreeClassifier()
clf = clf.fit(features, labels)#[prediction]
prediction = clf.predict([[130, "bumpy"]])print(prediction)
### 回答2:
以下是一个简单的决策树的Python代码示例:
```python
from sklearn import datasets
from sklearn.model_selection import train_test_split
from sklearn.tree import DecisionTreeClassifier
# 导入鸢尾花数据集
iris = datasets.load_iris()
X = iris.data
y = iris.target
# 将数据集分为训练集和测试集
X_train, X_test, y_train, y_test = train_test_split(X, y, test_size=0.2, random_state=42)
# 创建决策树模型
tree = DecisionTreeClassifier()
# 使用训练数据拟合模型
tree.fit(X_train, y_train)
# 使用测试数据进行预测
y_pred = tree.predict(X_test)
# 打印预测结果
print("预测结果:", y_pred)
```
上述代码首先导入所需的库,包括`sklearn`中的`datasets`模块用于加载鸢尾花数据集,`train_test_split`函数用于划分训练集和测试集,以及`DecisionTreeClassifier`类用于创建决策树分类器。
然后,加载鸢尾花数据集,将其特征数据保存为`X`,目标数据保存为`y`。
接下来,通过调用`train_test_split`函数将数据集划分为训练集和测试集,其中`test_size=0.2`表示测试集占总数据集的20%。
然后,创建决策树分类器对象,并将其保存在`tree`变量中。
接下来,使用训练数据来拟合(训练)决策树模型,即调用`fit`方法,传入训练数据`X_train`和对应的目标值`y_train`。
然后,使用测试数据进行预测,即调用`predict`方法,传入测试数据`X_test`。
最后,打印预测结果。
### 回答3:
以下是一个简单的决策树的Python代码示例:
```python
# 导入必要的库
from sklearn import datasets
from sklearn.model_selection import train_test_split
from sklearn.tree import DecisionTreeClassifier
from sklearn import metrics
# 加载数据集
dataset = datasets.load_iris()
X = dataset.data
y = dataset.target
# 将数据集划分为训练集和测试集
X_train, X_test, y_train, y_test = train_test_split(X, y, test_size=0.3, random_state=1)
# 创建决策树分类器
clf = DecisionTreeClassifier()
# 训练模型
clf.fit(X_train, y_train)
# 预测测试集
y_pred = clf.predict(X_test)
# 计算准确率
accuracy = metrics.accuracy_score(y_test, y_pred)
print("准确率:", accuracy)
```
这段代码使用了Scikit-learn库,首先加载了一个经典的鸢尾花数据集,然后将数据集拆分为训练集和测试集。接下来,创建了一个决策树分类器,并将训练集用于训练模型。最后,使用测试集进行预测,并计算准确率来评估模型的性能。
阅读全文