Hello.. Thanks for the repository... I have used vrae.fit for training on the dataset and vrae.transform() for tranforming test input to latent space. How to get the cluster assignments with regard to the test inputs... thanking you Praveen