Dear Authors,
I want to add one NN layer to DCRNN, STGCN, MTGNN before the final output. Do I just change the file in traffic_speed_predition folder?
I am just not very sure which method should I change. Do I rewrite the forward method in the class in each file?
Dear Authors,
I want to add one NN layer to DCRNN, STGCN, MTGNN before the final output. Do I just change the file in traffic_speed_predition folder?
I am just not very sure which method should I change. Do I rewrite the forward method in the class in each file?