├── .DS_Store ├── .gitignore ├── README.md ├── config.py ├── eval_metrics.py ├── hgcn_utils ├── .DS_Store ├── __init__.py ├── math_utils.py └── train_utils.py ├── layers ├── __init__.py └── hyp_layers.py ├── manifolds ├── .DS_Store ├── __init__.py ├── base.py └── hyperboloid.py ├── models ├── __init__.py ├── base_models.py └── encoders.py ├── rgd ├── __init__.py └── rsgd.py ├── run.py ├── utils ├── .DS_Store ├── __init__.py ├── data_generator.py ├── helper.py ├── log.py ├── preprocess.py └── sampler.py └── yelp20 └── user_item_list.pkl /.DS_Store: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/layer6ai-labs/HGCF/HEAD/.DS_Store -------------------------------------------------------------------------------- /.gitignore: -------------------------------------------------------------------------------- 1 | 2 | __pycache__/ 3 | -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/layer6ai-labs/HGCF/HEAD/README.md -------------------------------------------------------------------------------- /config.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/layer6ai-labs/HGCF/HEAD/config.py -------------------------------------------------------------------------------- /eval_metrics.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/layer6ai-labs/HGCF/HEAD/eval_metrics.py -------------------------------------------------------------------------------- /hgcn_utils/.DS_Store: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/layer6ai-labs/HGCF/HEAD/hgcn_utils/.DS_Store -------------------------------------------------------------------------------- /hgcn_utils/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/layer6ai-labs/HGCF/HEAD/hgcn_utils/__init__.py -------------------------------------------------------------------------------- /hgcn_utils/math_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/layer6ai-labs/HGCF/HEAD/hgcn_utils/math_utils.py -------------------------------------------------------------------------------- /hgcn_utils/train_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/layer6ai-labs/HGCF/HEAD/hgcn_utils/train_utils.py -------------------------------------------------------------------------------- /layers/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /layers/hyp_layers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/layer6ai-labs/HGCF/HEAD/layers/hyp_layers.py -------------------------------------------------------------------------------- /manifolds/.DS_Store: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/layer6ai-labs/HGCF/HEAD/manifolds/.DS_Store -------------------------------------------------------------------------------- /manifolds/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/layer6ai-labs/HGCF/HEAD/manifolds/__init__.py -------------------------------------------------------------------------------- /manifolds/base.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/layer6ai-labs/HGCF/HEAD/manifolds/base.py -------------------------------------------------------------------------------- /manifolds/hyperboloid.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/layer6ai-labs/HGCF/HEAD/manifolds/hyperboloid.py -------------------------------------------------------------------------------- /models/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /models/base_models.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/layer6ai-labs/HGCF/HEAD/models/base_models.py -------------------------------------------------------------------------------- /models/encoders.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/layer6ai-labs/HGCF/HEAD/models/encoders.py -------------------------------------------------------------------------------- /rgd/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /rgd/rsgd.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/layer6ai-labs/HGCF/HEAD/rgd/rsgd.py -------------------------------------------------------------------------------- /run.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/layer6ai-labs/HGCF/HEAD/run.py -------------------------------------------------------------------------------- /utils/.DS_Store: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/layer6ai-labs/HGCF/HEAD/utils/.DS_Store -------------------------------------------------------------------------------- /utils/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /utils/data_generator.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/layer6ai-labs/HGCF/HEAD/utils/data_generator.py -------------------------------------------------------------------------------- /utils/helper.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/layer6ai-labs/HGCF/HEAD/utils/helper.py -------------------------------------------------------------------------------- /utils/log.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/layer6ai-labs/HGCF/HEAD/utils/log.py -------------------------------------------------------------------------------- /utils/preprocess.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/layer6ai-labs/HGCF/HEAD/utils/preprocess.py -------------------------------------------------------------------------------- /utils/sampler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/layer6ai-labs/HGCF/HEAD/utils/sampler.py -------------------------------------------------------------------------------- /yelp20/user_item_list.pkl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/layer6ai-labs/HGCF/HEAD/yelp20/user_item_list.pkl --------------------------------------------------------------------------------