xixiareone/hyper-sagnn

hypergraph representation learning, graph neural network

Hyper-SAGNN: a self-attention based graph neural network for hypergraphs

This is an implementation of "Hyper-SAGNN: a self-attention based graph neural network for hypergraphs" (ICLR2020)

The datasets included in this repo are originally from DHNE (https://github.com/tadpole/DHNE)

Requirements

python >= 3.6.8

Tensorflow >= 1.0.0 (< 2.0.0)

Pytorch >= 1.0

Usage

To run the code:

cd Code
python main.py --data wordnet -f adj

Change the following arguments to reproduce corresponding results from the manuscript,

The --data argument can take "GPS", "drug", "MovieLens", "wordnet". This argument is case sensitive

The -f, --feature argument can take "adj" or "walk" represents encoder based approach and random walk based approach respectively.

Other arguments are as followed:

parser.add_argument('--dimensions', type=int, default=64,
                    help='Number of dimensions. Default is 64.')

parser.add_argument('-l', '--walk-length', type=int, default=40,
                    help='Length of walk per source. Default is 40.')

parser.add_argument('-r', '--num-walks', type=int, default=10,
                    help='Number of walks per source. Default is 10.')

parser.add_argument('-k', '--window-size', type=int, default=10,
                    help='Context size for optimization. Default is 10.')

Cite

If you want to cite our paper:

@inproceedings{
zhang2020hypersagnn,
title={Hyper-{SAGNN}: a self-attention based graph neural network for hypergraphs},
author={Zhang, Ruochi and Zou, Yuesong and Ma, Jian},
booktitle={International Conference on Learning Representations (ICLR)},
year={2020}
}