想用机器学习预测股价,但不知道该用 LSTM、GRU 还是 Transformer?Stock-Prediction-Models 项目汇集了 40 多种深度学习和机器学习模型,让你一次对比所有方案。
核心功能:
- 深度学习模型集合:包含 LSTM、GRU、Seq2seq、Attention、CNN 等 18 种神经网络架构
- 传统ML模型:集成 XGBoost、Random Forest、Adaboost 等经典机器学习算法
- 强化学习交易代理:提供 Q-learning、Policy Gradient、Actor-Critic 等 20+ 种 RL 交易策略
- 蒙特卡洛模拟:支持多种蒙特卡洛方法做风险评估和投资组合优化
- 浏览器可运行:提供 TensorFlow.js 版本,可直接在网页上传 CSV 进行预测
GitHub地址:https://github.com/huseinzol05/Stock-Prediction-Models
适合谁用:机器学习学习者、股票量化研究者、想对比不同预测模型效果的数据科学家