feat: 添加RUL-Mamba模型及相关组件
新增锂电池剩余使用寿命预测模型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
This commit is contained in:
@@ -0,0 +1,11 @@
|
||||
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
|
||||
Reference in New Issue
Block a user