-
Notifications
You must be signed in to change notification settings - Fork 0
Expand file tree
/
Copy pathargs.py
More file actions
46 lines (43 loc) · 1.75 KB
/
args.py
File metadata and controls
46 lines (43 loc) · 1.75 KB
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
class Args:
def __init__(self):
# General parameters
self.model_type = 'Informer' # Set to 'Informer' or 'RNN'
self.root_path = '/Users/scobi/Desktop/SCLoadForecasting/'
self.data_path = 'combinedDaytonData_fill.csv'
self.batch_size = 16
self.learning_rate = 0.001
self.epochs = 10
self.num_workers = 0
# RNN parameters
self.rnn_input_size = 4
self.rnn_hidden_size = 16
self.rnn_output_size = 1
self.rnn_num_layers = 1
# former parameters
self.features = "M"
self.target = "dayton_mw"
self.enc_in = 5 # Input features (temperature, precipitation, humidity, wind_speed)
self.dec_in = 5
self.c_out = 1 # Output feature (dayton_mw)
self.seq_len = 96
self.label_len = 48
self.pred_len = 24
self.d_model = 512 # Model dimensions
self.n_heads = 8 # Number of attention heads
self.e_layers = 2 # Number of encoder layers
self.d_layers = 1 # Number of decoder layers
self.d_ff = 2048 # Feed-forward network size
self.factor = 5 # ProbSparse factor
self.dropout = 0.1 # Dropout rate
self.embed = 'timeF' # Embedding type
self.embed_type = 0 # Embedding variant
self.freq = 'h' # Frequency ('h' for hourly)
self.activation = 'gelu' # Activation function
self.output_attention = False
self.distil = True # Whether to use distillation
self.moving_avg = 25 # Kernel size for moving average
self.moving_avg_kernel = 25
self.scale = True
self.timeenc = 0
self.train_only = False
self.individual = True # Set to True for feature-wise decomposition