dbn_traffic_prediction_keras icon indicating copy to clipboard operation
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