search

LibCity / Bigscity-LibCity-Datasets

Traffic data processing tools in LibCity
Apache License 2.0
186 stars 20 forks source link

Add one layer to DCRNN, STGCN, MTGNN #50

Open tianshubao1 opened 10 months ago

tianshubao1 commented 10 months ago

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?