[D] Using Variational Autoencoder for classification • r/MachineLearning

I am working on a project to classify a sample dataset but am looking to use a VAE. It has been suggested that I can use the latent feature, z (mu), as output, after training the autoencoder, with a simple neural networks to classify with the validation set. Could anyone please link me to any examples of the implementation, either from GitHub or other blog post with details, possibly using MNIST or other sample classification dataset?