dbn_traffic_prediction_pytorch

Introduction

This is a DBN regression model for sine wave prediction. The model is implemented in Pytorch 1.0.1.

Usage

Run main.py to get an instant prediction result. Model parameters and data can be modified according to your demand.

Reference

Special thanks to the following github repositories:

https://github.com/wmingwei/restricted-boltzmann-machine-deep-belief-network-deep-boltzmann-machine-in-pytorch

https://github.com/GabrielBianconi/pytorch-rbm

And here is a Keras version of DBN.

https://github.com/yyb1995/DBNTrafficflowPrediction