search

liyaguang / DCRNN

Implementation of Diffusion Convolutional Recurrent Neural Network in Tensorflow
MIT License
1.18k stars 391 forks source link

Tensorflow 2 for DCRNN models #77

Open samridhimaheshwari opened 1 year ago

samridhimaheshwari commented 1 year ago

Hello, Is there any update regarding updating the code to Tensorflow 2? a lot of Tensorflow functionality which is in this code no longer works. For example rnn_decoder is removed from tensorflow 1.x Thanks!

liyaguang commented 1 year ago

Thank you for reaching out. Unfortunately, I do not have the capacity to update the code to Tensorflow 2.0 at the moment. Besides, you are more than welcome to contribute to this, e.g., creating a branch with the TF 2.0 implementation. This would be greatly appreciated, as it would benefit others who are interested in using this code with Tensorflow 2.0.

zouguojian commented 1 year ago

Dear researcher, I have uploaded the codes that are adaptive to version 2.0; you can download these codes from this link: https://github.com/zouguojian/Traffic-speed-prediction/tree/main/3S-TBLN/baseline/DCRNN. (do not hesitate to provide the result you get)