dalian-university/加强训练/utils/Arg.py

class Arg():
    def __init__(self):
        self.input_dim = 18
        self.output_dim = 1
        self.input_seq_length = 16
        self.output_seq_length = 1
        self.d_model = 18
        self.nhead = 3
        self.num_layers = 4
        self.dropout = 0.05
        self.batch_size = 32
        self.epochs = 20
        self.time_split = 16
        self.power_max = 110
        self.power_min = 0
        self.hidden = 6