基于回归模型的销售预测
小P:小H,有没有什么好的办法预测下未来的销售额啊
小H:很多啊,简单的用统计中的一元/多元回归就好了,如果线性不明显,可以用机器学习训练预测
数据探索
导入相关库
# 导入库...# 初选回归模型
model_names = ['BayesianRidge', 'XGBR', 'ElasticNet', 'SVR', 'GBR'] # 不同模型的名称列表
model_br =...figsize=(10, 10))
for i, pre_y in enumerate(pre_y_list):
plt.subplot(len(pre_y_list)+1,1,i+1) # 子图6行...50, 100, 500],
'learning_rate': [0.05, 0.1, 0.3, 0.5],
'max_depth': [5, 6,...': 5, 'n_estimators': 50}
# 获取最佳训练模型
model_xgbr = model_gs.best_estimator_ # 获得交叉检验模型得出的最优模型对象
pre_y