| 123456789101112131415161718192021222324252627282930313233 |
- # GAT-LSTM TMP预测模型依赖包
- # Python 3.8+
- # 深度学习框架
- torch>=1.10.0
- torch-geometric>=2.0.0
- # 数据处理
- pandas>=1.3.0
- numpy>=1.21.0
- scikit-learn>=1.0.0
- # Web框架
- fastapi>=0.95.0
- uvicorn[standard]>=0.20.0
- pydantic>=1.10.0
- # 信号处理
- scipy>=1.7.0
- PyWavelets>=1.1.1
- # 工具库
- joblib>=1.1.0
- tqdm>=4.62.0
- colorlog>=6.6.0
- pyyaml>=6.0
- # 可选:CUDA支持(根据需要安装)
- # torch-cu117 # CUDA 11.7
- # torch-cu118 # CUDA 11.8
|