eason
79db6e5c96
新增锂电池剩余使用寿命预测模型RUL-Mamba，包含以下主要组件： 1. 添加Mamba模块作为核心时序建模组件 2. 实现特征注意力网络(FAN)和门控残差网络(GRN) 3. 新增数据预处理和归一化层 4. 添加模型训练和评估脚本 5. 补充README文档说明使用方法 6. 添加可视化辅助工具Helper_Plot.py 7. 实现多种时间序列处理层(Embedding、AutoCorrelation等) 8. 添加配置文件requirements.txt 9. 补充测试数据集TJU battery data
11 lines
199 B
Plaintext
11 lines
199 B
Plaintext
numpy==1.21.6
|
|
numba==0.55.1
|
|
matplotlib==3.3.4
|
|
scipy==1.8.0
|
|
statsmodels==0.13.5
|
|
pytorch-lightning==1.9.5
|
|
pytorch-forecasting==0.10.3
|
|
sympy==1.12.1
|
|
reformer_pytorch==1.4.4
|
|
openpyxl==3.1.5
|
|
einops==0.8.0 |