feat: add mamba and dynamic chunking related code and test code

This commit is contained in:
gameloader
2025-09-04 01:32:13 +00:00
parent 12cb7652cf
commit ef307a57e9
21 changed files with 4550 additions and 86 deletions

6
run.py
View File

@ -7,6 +7,7 @@ from exp.exp_imputation import Exp_Imputation
from exp.exp_short_term_forecasting import Exp_Short_Term_Forecast
from exp.exp_anomaly_detection import Exp_Anomaly_Detection
from exp.exp_classification import Exp_Classification
from exp.exp_dc_patchtst_classification import Exp_DC_PatchTST_Classification
from utils.print_args import print_args
import random
import numpy as np
@ -191,7 +192,10 @@ if __name__ == '__main__':
elif args.task_name == 'anomaly_detection':
Exp = Exp_Anomaly_Detection
elif args.task_name == 'classification':
Exp = Exp_Classification
if args.model == 'DC_PatchTST':
Exp = Exp_DC_PatchTST_Classification
else:
Exp = Exp_Classification
else:
Exp = Exp_Long_Term_Forecast