用Python写一段基于随机森林算法的销量预测代码
时间: 2023-05-13 18:05:29 浏览: 288
随机森林算法Python代码实现,ipynb格式
5星 · 资源好评率100%
当然可以,以下是一段基于随机森林算法的销量预测代码:
```python
# 导入必要的库
import pandas as pd
from sklearn.ensemble import RandomForestRegressor
from sklearn.model_selection import train_test_split
# 读取数据
data = pd.read_csv('sales_data.csv')
# 划分特征和标签
X = data.drop('sales', axis=1)
y = data['sales']
# 划分训练集和测试集
X_train, X_test, y_train, y_test = train_test_split(X, y, test_size=0.2, random_state=42)
# 定义随机森林模型
rf = RandomForestRegressor(n_estimators=100, random_state=42)
# 训练模型
rf.fit(X_train, y_train)
# 预测销量
y_pred = rf.predict(X_test)
# 输出模型评估结果
print('随机森林模型的R^2值为:', rf.score(X_test, y_test))
```
需要注意的是,这段代码仅供参考,具体的实现方式可能因数据集和业务场景的不同而有所差异。
阅读全文