dbn_traffic_prediction_keras
dbn_traffic_prediction_keras copied to clipboard
A traffic flow prediction model based on Scikit and Keras. The model contains a DBN and a NN.
dbn_traffic_prediction_keras
A traffic flow prediction model based on Scikit and Keras. The model contains a DBN and a NN. More functions will be added later.
Data source
http://www.cs.utexas.edu/~yzhang/research/AbileneTM
Link
And here is a Pytorch version of DBN prediction model.
https://github.com/yyb1995/dbn_traffic_prediction_pytorch