You are AllSet: A Multiset Function Framework for Hypergraph Neural Networks
This is the repo for our paper: You are AllSet: A Multiset Function Framework for Hypergraph Neural Networks. We prepared all codes and a subset of datasets used in our experiments.
All codes and script are in the folder src
, and a subset of raw data are provided in folder data
. To run the experiments, please go the the src
folder first.
Enviroment requirement:
This repo is tested with the following enviroment, higher version of torch PyG may also be compatible.
pytorch==1.4.0+cu100
torch-geometric==1.6.3
torch-scatter==2.0.4
Generate dataset from raw data.
To generate PyG or DGL dataset for training, please create the following three folders:
p2root: '../data/pyg_data/hypergraph_dataset_updated/'
p2raw: