yichentang97 / dbn_autoencoder_classifier Goto Github PK
View Code? Open in Web Editor NEWAn pytorch implementation of Deep Belief Network with sklearn compatibility for classification. The training process consists the pretraining of DBN, fine-tuning as an unrolled autoencoder-decoder, and supervised fine-tuning as a classifier.
License: MIT License