# 概念异动检测 ML 模块依赖
# 安装: pip install -r ml/requirements.txt

# PyTorch (根据 CUDA 版本选择)
# 5090 显卡需要 CUDA 12.x
# pip install torch --index-url https://download.pytorch.org/whl/cu124
torch>=2.0.0

# 数据处理
numpy>=1.24.0
pandas>=2.0.0
pyarrow>=14.0.0

# 数据库
clickhouse-driver>=0.2.6
elasticsearch>=7.0.0,<8.0.0
sqlalchemy>=2.0.0
pymysql>=1.1.0

# 训练工具
tqdm>=4.65.0

# 可选: 可视化
# matplotlib>=3.7.0
# tensorboard>=2.14.0