/acol-cluster

reproduce code for "https://arxiv.org/abs/1802.03063"

Primary LanguagePython

acol-cluster

reproduce code for "https://arxiv.org/abs/1802.03063"

run "python main.py", train the cnn model with mnist, and you can see cluster accuracy during training process on the command line

Todo List:

  • define model
  • transformation data for pseudo label
  • GAR regularization
  • train
  • get latent representation
  • acol pool
  • mnist test
  • svhn test