Meta-learning representations for clustering with infinite Gaussian mixture models
Paper
Tomoharu Iwata, "Meta-learning representations for clustering with infinite Gaussian mixture models," arXiv:2103.00694, 2021
arxiv
Codes
meta_gmm.tar.gz
Usage
-
Download Omniglot in "omniglot-py" directory.
% python download_omiglot.py
-
Generate datasets for meta-learning in "omniglot_tr0.6va0.2I10O50" directory.
% python crt_ominiglot_data.py
-
Run meta-learning.
% python train_meta_gmm.py --fn omniglot_tr0.6va0.2I10O50/i0.npz --out model
The codes were checked with Python 3.7, Pytorch 1.4, and Numpy 1.18.1.
License
LICENSE.txt