-
Notifications
You must be signed in to change notification settings - Fork 0
/
Copy pathconfig.json
42 lines (42 loc) · 1.17 KB
/
config.json
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
{
"data_path" : "./data/",
"filename" : "wtbdata_245days.csv",
"location_path" : "./data/sdwpf_baidukddcup2022_turb_location.CSV",
"thresh_distance": 1000,
"train_type" : "each",
"target" : "Patv",
"start_col" : 3,
"capacity" : 134,
"day_len" : 144,
"train_days" : 200,
"val_days" : 20,
"test_days" : 25,
"total_days" : 245,
"model_name" : "GTCN",
"input_size" : 10,
"hidden_size" : 20,
"output_len" : 24,
"input_len" : 48,
"num_layers" : 1,
"dropout" : 0.1,
"kernel_size" : 3,
"TCN_channels" : [128, 256, 512, 256],
"seg_len" : 6,
"nb_block" : 2,
"nb_chev_filter" : 64,
"nb_time_filter" : 64,
"time_strides" : 1,
"K" : 3,
"device" : "cuda",
"loss_fn" : "huber",
"batch_size" : 32,
"lr_rate" : 0.0005,
"max_epoch" : 50,
"patience" : 5,
"delta" : 0.001,
"lr_step_size" : 5,
"lr_gamma" : 0.9,
"p" : 1,
"d" : 1,
"q" : 1
}