-
Notifications
You must be signed in to change notification settings - Fork 12
Open
Description
I ran the command on latest small movielens dataset twice
python peagcn_solver_bpr.py --dataset=Movielens --dataset_name=latest-small --num_core=10 --num_feat_core=10 --sampling_strategy=unseen --entity_aware=false --dropout=0 --emb_dim=64 --repr_dim=16 --hidden_size=64 --meta_path_steps=2,2,2,2,2,2,2,2,2 --entity_aware_coff=0.1 --init_eval=true --gpu_idx=0 --runs=5 --epochs=30 --batch_size=1024 --save_every_epoch=5 --metapath_test=false
the first time is normal, but the second time I got this:
Dataset loaded!
Overall HR@5: 0.3803, HR@10: 0.5362, HR@15: 0.6329, HR@20: 0.7010, NDCG@5: 0.2503, NDCG@10: 0.2999, NDCG@15: 0.3246, NDCG@20: 0.3399, AUC: 0.8242, train loss: 259.2449, eval loss: 43.4085
It actually didn't running.
Metadata
Metadata
Assignees
Labels
No labels