├── LICENSE ├── README.md ├── config ├── __init__.py ├── config.py └── config.yaml ├── datasets ├── __init__.py ├── data_helper.py └── visual_data.py ├── doc └── pipline.png ├── models ├── HGNN.py ├── __init__.py └── layers.py ├── train.py └── utils └── hypergraph_utils.py /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/iMoonLab/HGNN/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/iMoonLab/HGNN/HEAD/README.md -------------------------------------------------------------------------------- /config/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/iMoonLab/HGNN/HEAD/config/__init__.py -------------------------------------------------------------------------------- /config/config.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/iMoonLab/HGNN/HEAD/config/config.py -------------------------------------------------------------------------------- /config/config.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/iMoonLab/HGNN/HEAD/config/config.yaml -------------------------------------------------------------------------------- /datasets/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/iMoonLab/HGNN/HEAD/datasets/__init__.py -------------------------------------------------------------------------------- /datasets/data_helper.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/iMoonLab/HGNN/HEAD/datasets/data_helper.py -------------------------------------------------------------------------------- /datasets/visual_data.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/iMoonLab/HGNN/HEAD/datasets/visual_data.py -------------------------------------------------------------------------------- /doc/pipline.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/iMoonLab/HGNN/HEAD/doc/pipline.png -------------------------------------------------------------------------------- /models/HGNN.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/iMoonLab/HGNN/HEAD/models/HGNN.py -------------------------------------------------------------------------------- /models/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/iMoonLab/HGNN/HEAD/models/__init__.py -------------------------------------------------------------------------------- /models/layers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/iMoonLab/HGNN/HEAD/models/layers.py -------------------------------------------------------------------------------- /train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/iMoonLab/HGNN/HEAD/train.py -------------------------------------------------------------------------------- /utils/hypergraph_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/iMoonLab/HGNN/HEAD/utils/hypergraph_utils.py --------------------------------------------------------------------------------