AI Stock Leader Engine

项目介绍

这是一个基于AI的股市龙头筛选引擎，旨在通过多维度数据分析（市场、新闻、基本面）和先进的机器学习模型（LSTM, XGBoost, Transformer）来识别市场龙头，预测短期走势，并优化交易策略。

核心目标

1. 识别市场龙头：通过板块动量和多因子筛选识别潜在龙头。
2. 预测短期走势：利用深度学习模型预测股价走势。
3. 优化交易策略：结合风险管理和自动优化算法提升交易回报。

目录结构

ai_stock_leader_engine/
├── README.md                           # 项目说明文档
├── requirements.txt                    # Python依赖包列表
├── config/                             # 配置文件
├── data/                               # 数据存储
├── src/                                # 源代码
│   ├── utils/                          # 通用工具
│   ├── data/                           # 数据获取与处理
│   ├── event_awareness/                # 事件感知
│   ├── sector_momentum/                # 板块动量
│   ├── leader_screening/               # 龙头筛选
│   ├── trend_prediction/               # 走势预测
│   ├── evaluation/                     # 回测与评估
│   ├── optimization/                   # 策略优化
│   ├── risk_management/                # 风险管理
│   ├── pipeline/                       # 执行管道
│   └── api/                            # 服务接口
└── notebooks/                          # 研究与实验

快速开始

1. 安装依赖

pip install -r requirements.txt

2. 配置环境

修改 config/ 目录下的配置文件，设置数据源API Key等信息。

3. 运行数据获取

python src/main.py --mode data_fetch

4. 训练模型

python src/main.py --mode train