python决策树模型训练代码
时间: 2023-08-11 10:04:53 浏览: 40
以下是一个简单的Python决策树模型训练的代码示例:
```python
from sklearn.tree import DecisionTreeClassifier
# 准备训练数据
X = \[\[1, 2\], \[3, 4\], \[5, 6\], \[7, 8\], \[9, 10\]\]
y = \[1, 2, 3, 4, 5\]
# 创建决策树模型
model = DecisionTreeClassifier(max_depth=2, random_state=0)
# 训练模型
model.fit(X, y)
# 使用模型进行预测
prediction = model.predict(\[\[9, 9\]\])
# 输出预测结果
print(prediction)
```
这段代码使用了`sklearn`库中的`DecisionTreeClassifier`类来创建一个决策树分类器模型。然后,通过调用`fit`方法来训练模型,传入训练数据`X`和对应的标签`y`。最后,使用`predict`方法对新的数据进行预测,输出预测结果。在这个例子中,预测结果为`\[4\]`。
#### 引用[.reference_title]
- *1* *2* [用通俗易懂的方式讲解:决策树模型及案例(Python 代码)](https://blog.csdn.net/qq_34160248/article/details/127170221)[target="_blank" data-report-click={"spm":"1018.2226.3001.9630","extra":{"utm_source":"vip_chatgpt_common_search_pc_result","utm_medium":"distribute.pc_search_result.none-task-cask-2~all~insert_cask~default-1-null.142^v91^insert_down1,239^v3^insert_chatgpt"}} ] [.reference_item]
- *3* [参数调优:K折交叉验证与GridSearch网格搜索](https://blog.csdn.net/m0_46388544/article/details/122718009)[target="_blank" data-report-click={"spm":"1018.2226.3001.9630","extra":{"utm_source":"vip_chatgpt_common_search_pc_result","utm_medium":"distribute.pc_search_result.none-task-cask-2~all~insert_cask~default-1-null.142^v91^insert_down1,239^v3^insert_chatgpt"}} ] [.reference_item]
[ .reference_list ]
相关推荐

















