Link of Pytorch installation is here: https://pytorch.org/. If Pytorch is already installed then this is not necessary.
Link of Pytorch Lightning: https://github.com/PyTorchLightning/pytorch-lightning.
pip install pytorch-lightning
pip install ipynb
X
- The feature
y
- labels, not necessary only useful if used for visualization
dim
- The output dimension of the transformed feature representation, change this accordingly
latent
- Latent representation of the mean and variance of the disitribution
Hidden
- Number of neurons in hidden layers
Iteration
- How many iterations you want to run
visualize
- Visualize encoder and latent representation