requirements.txt 464 B

123456789101112131415161718192021222324252627282930313233
  1. # GAT-LSTM TMP预测模型依赖包
  2. # Python 3.8+
  3. # 深度学习框架
  4. torch>=1.10.0
  5. torch-geometric>=2.0.0
  6. # 数据处理
  7. pandas>=1.3.0
  8. numpy>=1.21.0
  9. scikit-learn>=1.0.0
  10. # Web框架
  11. fastapi>=0.95.0
  12. uvicorn[standard]>=0.20.0
  13. pydantic>=1.10.0
  14. # 信号处理
  15. scipy>=1.7.0
  16. PyWavelets>=1.1.1
  17. # 工具库
  18. joblib>=1.1.0
  19. tqdm>=4.62.0
  20. colorlog>=6.6.0
  21. pyyaml>=6.0
  22. # 可选:CUDA支持(根据需要安装)
  23. # torch-cu117 # CUDA 11.7
  24. # torch-cu118 # CUDA 11.8