- Train and test sets of Gowalla, Yelp2020 and Amazon-book are located in here.
Run program with Gowalla dataset with bucket size 500, 2 compositional meta-embeddings/entity on cuda device 0:
python3 engine.py --dataset_name gowalla --num_clusters 500 --num_composition_centroid 2 --device_id 0
-
Gowalla
Hyperparam Value GCN layer 3 l2 penalty factor 5 lr 1e-3 assignment update freq $m$ every epoch #composition embs per entity $t$ 2 #clusters $c$ 500 init anchor weight $w^*$ 0.5 -
Yelp2020
Hyperparam Value GCN layer 4 l2 penalty factor 5 lr 1e-3 assignment update freq $m$ every epoch #composition embs per entity $t$ 2 #clusters $c$ 500 init anchor weight $w^*$ 0.6 -
Amazon-book
Hyperparam Value GCN layer 4 l2 penalty factor 5 lr 1e-3 assignment update freq $m$ every epoch #composition embs per entity $t$ 2 #clusters $c$ 500 init anchor weight $w^*$ 0.9