Pretrained models are available in cache/ckpt
.
We also provide cached posterior z samples in cache
. They can be used to evaluate the clustering and semi-supervised classification performance.
python train_sri.py
python unsupervised_cluster.py
python semi_sup_classification.py
python train_sri.py --eval_only False
Feel free to contact Bo Pang (bopang@ucla.edu) or Erik Nijkamp (enijkamp@ucla.edu) for questions.