parent
d1bc922ee7
commit
b94291a27c
@ -0,0 +1,28 @@
|
||||
[Data]
|
||||
adj_filename = ./data/METR_LA/distance_LA.csv
|
||||
graph_signal_matrix_filename = ./data/METR_LA/METR_LA.npz
|
||||
num_of_vertices = 207
|
||||
points_per_hour = 12
|
||||
num_for_predict = 12
|
||||
len_input = 12
|
||||
dataset_name = METR_LA
|
||||
|
||||
[Training]
|
||||
ctx = 0
|
||||
in_channels = 1
|
||||
nb_block = 2
|
||||
K = 3
|
||||
nb_chev_filter = 64
|
||||
nb_time_filter = 64
|
||||
batch_size = 16
|
||||
model_name = astgcn_r
|
||||
dataset_name = METR_LA
|
||||
num_of_weeks = 0
|
||||
num_of_days = 0
|
||||
num_of_hours = 1
|
||||
start_epoch = 0
|
||||
epochs = 100
|
||||
learning_rate = 0.001
|
||||
loss_function = masked_mae
|
||||
metric_method=mask
|
||||
missing_value=0.0
|
Loading…
Reference in new issue