model = DecisionTreeClassifier() model.fit(data_train, label_train) print('Decision Tree Model Trained') y_pred = model.predict(data_test)改成随机森林
时间: 2024-02-22 08:55:25 浏览: 94
RF_Class_C.rar_classRF_train_随机森林 分类_随机森林matlab_随机森林分类_随机森林算法
5星 · 资源好评率100%
将决策树模型改为随机森林模型,只需要将代码中的 `DecisionTreeClassifier()` 修改为 `RandomForestClassifier()` 即可,具体代码如下:
```
model = RandomForestClassifier()
model.fit(data_train, label_train)
print('Random Forest Model Trained')
y_pred = model.predict(data_test)
```
注意,随机森林模型有一些超参数需要调节,例如决策树的数量、每棵树的最大深度等。你可以使用交叉验证等技术来选择最佳的超参数组合。
阅读全文