diff --git a/name/PEMS04_astgcn.conf b/name/PEMS04_astgcn.conf new file mode 100644 index 0000000..cb89e09 --- /dev/null +++ b/name/PEMS04_astgcn.conf @@ -0,0 +1,28 @@ +[Data] +adj_filename = ./data/PEMS04/distance.csv +graph_signal_matrix_filename = ./data/PEMS04/PEMS04.npz +num_of_vertices = 307 +points_per_hour = 12 +num_for_predict = 12 +len_input = 12 +dataset_name = PEMS04 + +[Training] +ctx = 0 +in_channels = 1 +nb_block = 2 +K = 3 +nb_chev_filter = 64 +nb_time_filter = 64 +batch_size = 64 +model_name = astgcn_r +dataset_name = PEMS04 +num_of_weeks = 0 +num_of_days = 0 +num_of_hours = 1 +start_epoch = 0 +epochs = 80 +learning_rate = 0.001 +loss_function = masked_mae +metric_method = mask +missing_value=0.0