feat: add mamba and dynamic chunking related code and test code
This commit is contained in:
6
run.py
6
run.py
@ -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
|
||||
|
||||
|
Reference in New Issue
Block a user