项目简介
本项目是基于深度学习算法构建的股票预测系统,借助分析历史交易数据来预测股票的涨跌趋势。系统运用Transformer和Time2Vec等先进技术,结合自研的ModestCattle模型,把股票预测问题当作分类问题处理,以此提升预测的准确性与合理性。
项目的主要特性和功能
- 深度学习模型:基于Transformer和Time2Vec的ModestCattle模型,结合注意力机制与时间特征提取,提高预测精度。
- 分类问题处理:将股票预测视为分类问题,预测涨跌概率,而非具体股价数值,更具合理性与实用性。
- 多源数据融合:整合个股K线、大盘走势、新闻、政策和散户评论等多源信息进行综合分析。
- 模型训练与评估:提供从数据获取、模型训练到评估的完整流程,支持日K、分时、大盘等多维度数据的学习。
- 开源与交流:项目遵循Heartfelt Goodness开源协议,鼓励用户学习交流和改进。
安装使用步骤
原始版本(仅支持日K学习)
- 获取股票代码:运行
get_code.py
,获取当前市场存在的股票代码。 - 获取训练数据:运行
get_data.py
,通过股票代码获取训练样本数据。 - 模型训练:运行
train_with_?.py
(如train_with_transformer.py
)进行模型训练。 - 模型评估:运行
evel_with_?.py
(如evel_with_transformer.py
)获取市场标的的预测结果。
SQL版本(支持多编码学习)
- 数据库导入:下载并导入
ai_stock.sql
数据库文件。 - 数据下载:运行
get_data_to_sql.py
和get_trick_to_sql.py
,下载日K和分时数据,并修改数据库连接配置。 - 模型训练:运行
train_with_Transfomer_from_sql.py
进行模型训练。 - 模型评估:运行
evel_with_transformer_sql.py
查看预测效果。
注意事项
- 本项目预测结果仅供参考,不构成投资建议。
- 使用项目代码时,需确保遵守相关法律法规,不用于非法用途。
- 模型训练效果受数据质量、模型参数、训练策略等多种因素影响,需根据实际情况调整和优化。
下载地址
点击下载 【提取码: 4003】【解压密码: www.makuang.net】