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

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

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