🔠项目运行地址(免费):点击跳转
👍项目推荐
欢迎查看更多项目:🏆Machine Learning实战指南
📃项目背景
本数据集收录了Netflix自2002年开始至2022年的股价数据,数据来源为Yahoo Finance。
1.数据集理解
科代表提前总结:
- 该数据集维度为(5044,7),7个特征当中,有6个数值型特征和1个非数值型(object)特征;
- 数据集无缺失值和重复值;
- 数据集中的“Date”特征是非数值型(object)特征,后期需要转为时间戳格式。
2.数据可视化
2.1 奈飞日线行情(蜡烛图)
2.2 奈飞的收盘价移动平均线
3.时序模型构建及预测
笔者在这里将使用NeuralProphet进行训练,NeuralProphet仅支持两列特征作为输入:时间戳+数值。所以在这里,笔者先取出“时间戳”和“收盘价”作为后续预测。
3.1 baseline版本
不难看出,用NeuralProphet的默认参数进行预测的结果并不是很理想,在训练集和测试集上表现的不是很好,应该算是“欠拟合”,我们接下来尝试通过optuna对NeuralProphet进行优化调参。
3.1.1 训练集预测结果
3.1.2 验证集预测结果
3.2 optuna调参版本
经过optuna调优后的neuralprophet,可以很好的预测奈飞的股价趋势。
3.2.1 训练集预测结果
3.2.2 验证集预测结果