def get_dataset_parameters(opt):
    """Prepare specific parameters for different datasets"""
    dataset2num = {
        'elect': 370,
        'flow': 1083,
        'wind': 29
    }
    dataset2covariate = {
        'elect':3,
        'flow': 3,
        'wind': 3
    }
    dataset2input_len = {
        'elect':169,
        'flow': 192,
        'wind': 192
    }
    dataset2ignore_zero = {
        'elect': True,
        'flow': True,
        'wind': False
    }

    opt.num_seq = dataset2num[opt.dataset]
    opt.covariate_size = dataset2covariate[opt.dataset]
    opt.input_size = dataset2input_len[opt.dataset]
    opt.ignore_zero = dataset2ignore_zero[opt.dataset]
    return opt
Prepare specific parameters for different datasets
PatchTST_supervised/Formers/Pyraformer/single_step_main.py