python随机森林预测
时间: 2023-05-31 18:03:33 浏览: 104
随机森林是一种强大的机器学习算法,可以用于分类和回归问题。在Python中,可以使用scikit-learn库中的RandomForestClassifier和RandomForestRegressor类来实现随机森林分类和回归。
下面是一个示例代码,用于使用随机森林进行二元分类:
``` python
from sklearn.ensemble import RandomForestClassifier
from sklearn.datasets import make_classification
# 创建一个二元分类数据集
X, y = make_classification(n_samples=1000, n_features=4, n_informative=2, n_redundant=0, random_state=0, shuffle=False)
# 创建一个随机森林分类器
clf = RandomForestClassifier(n_estimators=100, max_depth=2, random_state=0)
# 使用数据集拟合分类器
clf.fit(X, y)
# 预测新数据
print(clf.predict([[0, 0, 0, 0]]))
```
上面的代码首先创建一个二元分类数据集,然后使用RandomForestClassifier类创建一个随机森林分类器。接下来,使用数据集拟合分类器,并使用predict方法预测新数据的分类。
类似地,下面是一个示例代码,用于使用随机森林进行回归:
``` python
from sklearn.ensemble import RandomForestRegressor
from sklearn.datasets import make_regression
# 创建一个回归数据集
X, y = make_regression(n_samples=1000, n_features=4, n_informative=2, random_state=0, shuffle=False)
# 创建一个随机森林回归器
regr = RandomForestRegressor(n_estimators=100, max_depth=2, random_state=0)
# 使用数据集拟合回归器
regr.fit(X, y)
# 预测新数据
print(regr.predict([[0, 0, 0, 0]]))
```
上面的代码首先创建一个回归数据集,然后使用RandomForestRegressor类创建一个随机森林回归器。接下来,使用数据集拟合回归器,并使用predict方法预测新数据的回归值。
阅读全文