├── LICENSE ├── README.md ├── Results.png ├── experiments ├── cfkg_solver_bpr.py ├── checkpoint │ ├── data │ │ └── Movielenslatest-small │ │ │ └── processed │ │ │ └── ml_latest-small_core_10_type_hete.pkl │ └── weights │ │ └── Movielenslatest-small │ │ ├── PEAGAT │ │ └── BPR │ │ │ ├── {'model_type': 'Graph', 'if_use_features': False, 'emb_dim': 64, 'hidden_size': 64, 'repr_dim': 16, 'dropout': 0, 'num_heads': 1, 'meta_path_steps': 9, 'channel_aggr': 'att', 'entity_aware': False, 'entity_aware_coff': 0.1} │ │ │ └── run_1 │ │ │ │ └── latest.pkl │ │ │ └── {'model_type': 'Graph', 'if_use_features': False, 'emb_dim': 64, 'hidden_size': 64, 'repr_dim': 16, 'dropout': 0, 'num_heads': 1, 'meta_path_steps': 9, 'channel_aggr': 'att', 'entity_aware': True, 'entity_aware_coff': 0.1} │ │ │ └── run_1 │ │ │ └── latest.pkl │ │ ├── PEAGCN │ │ └── BPR │ │ │ ├── {'model_type': 'Graph', 'if_use_features': False, 'emb_dim': 64, 'hidden_size': 64, 'repr_dim': 16, 'dropout': 0, 'meta_path_steps': 9, 'channel_aggr': 'att', 'entity_aware': False, 'entity_aware_coff': 0.1} │ │ │ └── run_1 │ │ │ │ └── latest.pkl │ │ │ └── {'model_type': 'Graph', 'if_use_features': False, 'emb_dim': 64, 'hidden_size': 64, 'repr_dim': 16, 'dropout': 0, 'meta_path_steps': 9, 'channel_aggr': 'att', 'entity_aware': True, 'entity_aware_coff': 0.1} │ │ │ └── run_1 │ │ │ └── latest.pkl │ │ └── PEASage │ │ └── BPR │ │ ├── {'model_type': 'Graph', 'if_use_features': False, 'emb_dim': 64, 'hidden_size': 64, 'repr_dim': 16, 'dropout': 0, 'meta_path_steps': 9, 'channel_aggr': 'att', 'entity_aware': False, 'entity_aware_coff': 0.1} │ │ └── run_1 │ │ │ └── latest.pkl │ │ └── {'model_type': 'Graph', 'if_use_features': False, 'emb_dim': 64, 'hidden_size': 64, 'repr_dim': 16, 'dropout': 0, 'meta_path_steps': 9, 'channel_aggr': 'att', 'entity_aware': True, 'entity_aware_coff': 0.1} │ │ └── run_1 │ │ └── latest.pkl ├── herec_solver_bpr.py ├── kgat_solver_bpr.py ├── kgcn_solver_bpr.py ├── lgc_solver_bpr.py ├── metapath2vec_solver_bpr.py ├── multi_gccf_solver_bpr.py ├── nfm_solver_bce.py ├── ngcf_solver_bpr.py ├── peagat_solver_bpr.py ├── peagcn_solver_bpr.py ├── peasage_solver_bpr.py ├── scripts │ ├── script_movielens_25m.ps1 │ ├── script_movielens_latest-small.ps1 │ └── script_yelp.ps1 └── xdfm_solver_bpr.py ├── graph_recsys_benchmark ├── __init__.py ├── datasets │ ├── __init__.py │ ├── dataset.py │ ├── movielens.py │ └── yelp.py ├── models │ ├── __init__.py │ ├── base.py │ ├── cfkg.py │ ├── herec.py │ ├── kgat.py │ ├── kgcn.py │ ├── lgc.py │ ├── metapath2vec.py │ ├── multi_gccf.py │ ├── nfm.py │ ├── ngcf.py │ ├── peagat.py │ ├── peagcn.py │ ├── peasage.py │ ├── walk.py │ └── xdfm.py ├── nn │ ├── __init__.py │ ├── kgat_conv.py │ ├── kgcn_conv.py │ ├── multi_gccf_conv.py │ ├── ngcf_conv.py │ └── sum_aggregator_conv.py ├── parser │ ├── __init__.py │ ├── ml.py │ └── yelp.py ├── solvers.py └── utils │ ├── __init__.py │ ├── general_utils.py │ └── rec_utils.py ├── images └── framework.png ├── requirements.txt └── setup.py /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/blindsubmission1/PEAGNN/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/blindsubmission1/PEAGNN/HEAD/README.md -------------------------------------------------------------------------------- /Results.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/blindsubmission1/PEAGNN/HEAD/Results.png -------------------------------------------------------------------------------- /experiments/cfkg_solver_bpr.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/blindsubmission1/PEAGNN/HEAD/experiments/cfkg_solver_bpr.py -------------------------------------------------------------------------------- /experiments/checkpoint/data/Movielenslatest-small/processed/ml_latest-small_core_10_type_hete.pkl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/blindsubmission1/PEAGNN/HEAD/experiments/checkpoint/data/Movielenslatest-small/processed/ml_latest-small_core_10_type_hete.pkl -------------------------------------------------------------------------------- /experiments/checkpoint/weights/Movielenslatest-small/PEAGAT/BPR/{'model_type': 'Graph', 'if_use_features': False, 'emb_dim': 64, 'hidden_size': 64, 'repr_dim': 16, 'dropout': 0, 'num_heads': 1, 'meta_path_steps': 9, 'channel_aggr': 'att', 'entity_aware': False, 'entity_aware_coff': 0.1}/run_1/latest.pkl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/blindsubmission1/PEAGNN/HEAD/experiments/checkpoint/weights/Movielenslatest-small/PEAGAT/BPR/{'model_type': 'Graph', 'if_use_features': False, 'emb_dim': 64, 'hidden_size': 64, 'repr_dim': 16, 'dropout': 0, 'num_heads': 1, 'meta_path_steps': 9, 'channel_aggr': 'att', 'entity_aware': False, 'entity_aware_coff': 0.1}/run_1/latest.pkl -------------------------------------------------------------------------------- /experiments/checkpoint/weights/Movielenslatest-small/PEAGAT/BPR/{'model_type': 'Graph', 'if_use_features': False, 'emb_dim': 64, 'hidden_size': 64, 'repr_dim': 16, 'dropout': 0, 'num_heads': 1, 'meta_path_steps': 9, 'channel_aggr': 'att', 'entity_aware': True, 'entity_aware_coff': 0.1}/run_1/latest.pkl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/blindsubmission1/PEAGNN/HEAD/experiments/checkpoint/weights/Movielenslatest-small/PEAGAT/BPR/{'model_type': 'Graph', 'if_use_features': False, 'emb_dim': 64, 'hidden_size': 64, 'repr_dim': 16, 'dropout': 0, 'num_heads': 1, 'meta_path_steps': 9, 'channel_aggr': 'att', 'entity_aware': True, 'entity_aware_coff': 0.1}/run_1/latest.pkl -------------------------------------------------------------------------------- /experiments/checkpoint/weights/Movielenslatest-small/PEAGCN/BPR/{'model_type': 'Graph', 'if_use_features': False, 'emb_dim': 64, 'hidden_size': 64, 'repr_dim': 16, 'dropout': 0, 'meta_path_steps': 9, 'channel_aggr': 'att', 'entity_aware': False, 'entity_aware_coff': 0.1}/run_1/latest.pkl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/blindsubmission1/PEAGNN/HEAD/experiments/checkpoint/weights/Movielenslatest-small/PEAGCN/BPR/{'model_type': 'Graph', 'if_use_features': False, 'emb_dim': 64, 'hidden_size': 64, 'repr_dim': 16, 'dropout': 0, 'meta_path_steps': 9, 'channel_aggr': 'att', 'entity_aware': False, 'entity_aware_coff': 0.1}/run_1/latest.pkl -------------------------------------------------------------------------------- /experiments/checkpoint/weights/Movielenslatest-small/PEAGCN/BPR/{'model_type': 'Graph', 'if_use_features': False, 'emb_dim': 64, 'hidden_size': 64, 'repr_dim': 16, 'dropout': 0, 'meta_path_steps': 9, 'channel_aggr': 'att', 'entity_aware': True, 'entity_aware_coff': 0.1}/run_1/latest.pkl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/blindsubmission1/PEAGNN/HEAD/experiments/checkpoint/weights/Movielenslatest-small/PEAGCN/BPR/{'model_type': 'Graph', 'if_use_features': False, 'emb_dim': 64, 'hidden_size': 64, 'repr_dim': 16, 'dropout': 0, 'meta_path_steps': 9, 'channel_aggr': 'att', 'entity_aware': True, 'entity_aware_coff': 0.1}/run_1/latest.pkl -------------------------------------------------------------------------------- /experiments/checkpoint/weights/Movielenslatest-small/PEASage/BPR/{'model_type': 'Graph', 'if_use_features': False, 'emb_dim': 64, 'hidden_size': 64, 'repr_dim': 16, 'dropout': 0, 'meta_path_steps': 9, 'channel_aggr': 'att', 'entity_aware': False, 'entity_aware_coff': 0.1}/run_1/latest.pkl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/blindsubmission1/PEAGNN/HEAD/experiments/checkpoint/weights/Movielenslatest-small/PEASage/BPR/{'model_type': 'Graph', 'if_use_features': False, 'emb_dim': 64, 'hidden_size': 64, 'repr_dim': 16, 'dropout': 0, 'meta_path_steps': 9, 'channel_aggr': 'att', 'entity_aware': False, 'entity_aware_coff': 0.1}/run_1/latest.pkl -------------------------------------------------------------------------------- /experiments/checkpoint/weights/Movielenslatest-small/PEASage/BPR/{'model_type': 'Graph', 'if_use_features': False, 'emb_dim': 64, 'hidden_size': 64, 'repr_dim': 16, 'dropout': 0, 'meta_path_steps': 9, 'channel_aggr': 'att', 'entity_aware': True, 'entity_aware_coff': 0.1}/run_1/latest.pkl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/blindsubmission1/PEAGNN/HEAD/experiments/checkpoint/weights/Movielenslatest-small/PEASage/BPR/{'model_type': 'Graph', 'if_use_features': False, 'emb_dim': 64, 'hidden_size': 64, 'repr_dim': 16, 'dropout': 0, 'meta_path_steps': 9, 'channel_aggr': 'att', 'entity_aware': True, 'entity_aware_coff': 0.1}/run_1/latest.pkl -------------------------------------------------------------------------------- /experiments/herec_solver_bpr.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/blindsubmission1/PEAGNN/HEAD/experiments/herec_solver_bpr.py -------------------------------------------------------------------------------- /experiments/kgat_solver_bpr.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/blindsubmission1/PEAGNN/HEAD/experiments/kgat_solver_bpr.py -------------------------------------------------------------------------------- /experiments/kgcn_solver_bpr.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/blindsubmission1/PEAGNN/HEAD/experiments/kgcn_solver_bpr.py -------------------------------------------------------------------------------- /experiments/lgc_solver_bpr.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/blindsubmission1/PEAGNN/HEAD/experiments/lgc_solver_bpr.py -------------------------------------------------------------------------------- /experiments/metapath2vec_solver_bpr.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/blindsubmission1/PEAGNN/HEAD/experiments/metapath2vec_solver_bpr.py -------------------------------------------------------------------------------- /experiments/multi_gccf_solver_bpr.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/blindsubmission1/PEAGNN/HEAD/experiments/multi_gccf_solver_bpr.py -------------------------------------------------------------------------------- /experiments/nfm_solver_bce.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/blindsubmission1/PEAGNN/HEAD/experiments/nfm_solver_bce.py -------------------------------------------------------------------------------- /experiments/ngcf_solver_bpr.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/blindsubmission1/PEAGNN/HEAD/experiments/ngcf_solver_bpr.py -------------------------------------------------------------------------------- /experiments/peagat_solver_bpr.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/blindsubmission1/PEAGNN/HEAD/experiments/peagat_solver_bpr.py -------------------------------------------------------------------------------- /experiments/peagcn_solver_bpr.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/blindsubmission1/PEAGNN/HEAD/experiments/peagcn_solver_bpr.py -------------------------------------------------------------------------------- /experiments/peasage_solver_bpr.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/blindsubmission1/PEAGNN/HEAD/experiments/peasage_solver_bpr.py -------------------------------------------------------------------------------- /experiments/scripts/script_movielens_25m.ps1: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/blindsubmission1/PEAGNN/HEAD/experiments/scripts/script_movielens_25m.ps1 -------------------------------------------------------------------------------- /experiments/scripts/script_movielens_latest-small.ps1: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/blindsubmission1/PEAGNN/HEAD/experiments/scripts/script_movielens_latest-small.ps1 -------------------------------------------------------------------------------- /experiments/scripts/script_yelp.ps1: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/blindsubmission1/PEAGNN/HEAD/experiments/scripts/script_yelp.ps1 -------------------------------------------------------------------------------- /experiments/xdfm_solver_bpr.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/blindsubmission1/PEAGNN/HEAD/experiments/xdfm_solver_bpr.py -------------------------------------------------------------------------------- /graph_recsys_benchmark/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/blindsubmission1/PEAGNN/HEAD/graph_recsys_benchmark/__init__.py -------------------------------------------------------------------------------- /graph_recsys_benchmark/datasets/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/blindsubmission1/PEAGNN/HEAD/graph_recsys_benchmark/datasets/__init__.py -------------------------------------------------------------------------------- /graph_recsys_benchmark/datasets/dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/blindsubmission1/PEAGNN/HEAD/graph_recsys_benchmark/datasets/dataset.py -------------------------------------------------------------------------------- /graph_recsys_benchmark/datasets/movielens.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/blindsubmission1/PEAGNN/HEAD/graph_recsys_benchmark/datasets/movielens.py -------------------------------------------------------------------------------- /graph_recsys_benchmark/datasets/yelp.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/blindsubmission1/PEAGNN/HEAD/graph_recsys_benchmark/datasets/yelp.py -------------------------------------------------------------------------------- /graph_recsys_benchmark/models/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/blindsubmission1/PEAGNN/HEAD/graph_recsys_benchmark/models/__init__.py -------------------------------------------------------------------------------- /graph_recsys_benchmark/models/base.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/blindsubmission1/PEAGNN/HEAD/graph_recsys_benchmark/models/base.py -------------------------------------------------------------------------------- /graph_recsys_benchmark/models/cfkg.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/blindsubmission1/PEAGNN/HEAD/graph_recsys_benchmark/models/cfkg.py -------------------------------------------------------------------------------- /graph_recsys_benchmark/models/herec.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/blindsubmission1/PEAGNN/HEAD/graph_recsys_benchmark/models/herec.py -------------------------------------------------------------------------------- /graph_recsys_benchmark/models/kgat.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/blindsubmission1/PEAGNN/HEAD/graph_recsys_benchmark/models/kgat.py -------------------------------------------------------------------------------- /graph_recsys_benchmark/models/kgcn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/blindsubmission1/PEAGNN/HEAD/graph_recsys_benchmark/models/kgcn.py -------------------------------------------------------------------------------- /graph_recsys_benchmark/models/lgc.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/blindsubmission1/PEAGNN/HEAD/graph_recsys_benchmark/models/lgc.py -------------------------------------------------------------------------------- /graph_recsys_benchmark/models/metapath2vec.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/blindsubmission1/PEAGNN/HEAD/graph_recsys_benchmark/models/metapath2vec.py -------------------------------------------------------------------------------- /graph_recsys_benchmark/models/multi_gccf.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/blindsubmission1/PEAGNN/HEAD/graph_recsys_benchmark/models/multi_gccf.py -------------------------------------------------------------------------------- /graph_recsys_benchmark/models/nfm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/blindsubmission1/PEAGNN/HEAD/graph_recsys_benchmark/models/nfm.py -------------------------------------------------------------------------------- /graph_recsys_benchmark/models/ngcf.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/blindsubmission1/PEAGNN/HEAD/graph_recsys_benchmark/models/ngcf.py -------------------------------------------------------------------------------- /graph_recsys_benchmark/models/peagat.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/blindsubmission1/PEAGNN/HEAD/graph_recsys_benchmark/models/peagat.py -------------------------------------------------------------------------------- /graph_recsys_benchmark/models/peagcn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/blindsubmission1/PEAGNN/HEAD/graph_recsys_benchmark/models/peagcn.py -------------------------------------------------------------------------------- /graph_recsys_benchmark/models/peasage.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/blindsubmission1/PEAGNN/HEAD/graph_recsys_benchmark/models/peasage.py -------------------------------------------------------------------------------- /graph_recsys_benchmark/models/walk.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/blindsubmission1/PEAGNN/HEAD/graph_recsys_benchmark/models/walk.py -------------------------------------------------------------------------------- /graph_recsys_benchmark/models/xdfm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/blindsubmission1/PEAGNN/HEAD/graph_recsys_benchmark/models/xdfm.py -------------------------------------------------------------------------------- /graph_recsys_benchmark/nn/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/blindsubmission1/PEAGNN/HEAD/graph_recsys_benchmark/nn/__init__.py -------------------------------------------------------------------------------- /graph_recsys_benchmark/nn/kgat_conv.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/blindsubmission1/PEAGNN/HEAD/graph_recsys_benchmark/nn/kgat_conv.py -------------------------------------------------------------------------------- /graph_recsys_benchmark/nn/kgcn_conv.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/blindsubmission1/PEAGNN/HEAD/graph_recsys_benchmark/nn/kgcn_conv.py -------------------------------------------------------------------------------- /graph_recsys_benchmark/nn/multi_gccf_conv.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/blindsubmission1/PEAGNN/HEAD/graph_recsys_benchmark/nn/multi_gccf_conv.py -------------------------------------------------------------------------------- /graph_recsys_benchmark/nn/ngcf_conv.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/blindsubmission1/PEAGNN/HEAD/graph_recsys_benchmark/nn/ngcf_conv.py -------------------------------------------------------------------------------- /graph_recsys_benchmark/nn/sum_aggregator_conv.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/blindsubmission1/PEAGNN/HEAD/graph_recsys_benchmark/nn/sum_aggregator_conv.py -------------------------------------------------------------------------------- /graph_recsys_benchmark/parser/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/blindsubmission1/PEAGNN/HEAD/graph_recsys_benchmark/parser/__init__.py -------------------------------------------------------------------------------- /graph_recsys_benchmark/parser/ml.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/blindsubmission1/PEAGNN/HEAD/graph_recsys_benchmark/parser/ml.py -------------------------------------------------------------------------------- /graph_recsys_benchmark/parser/yelp.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/blindsubmission1/PEAGNN/HEAD/graph_recsys_benchmark/parser/yelp.py -------------------------------------------------------------------------------- /graph_recsys_benchmark/solvers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/blindsubmission1/PEAGNN/HEAD/graph_recsys_benchmark/solvers.py -------------------------------------------------------------------------------- /graph_recsys_benchmark/utils/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/blindsubmission1/PEAGNN/HEAD/graph_recsys_benchmark/utils/__init__.py -------------------------------------------------------------------------------- /graph_recsys_benchmark/utils/general_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/blindsubmission1/PEAGNN/HEAD/graph_recsys_benchmark/utils/general_utils.py -------------------------------------------------------------------------------- /graph_recsys_benchmark/utils/rec_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/blindsubmission1/PEAGNN/HEAD/graph_recsys_benchmark/utils/rec_utils.py -------------------------------------------------------------------------------- /images/framework.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/blindsubmission1/PEAGNN/HEAD/images/framework.png -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/blindsubmission1/PEAGNN/HEAD/requirements.txt -------------------------------------------------------------------------------- /setup.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/blindsubmission1/PEAGNN/HEAD/setup.py --------------------------------------------------------------------------------