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:

 

 

 

To finish reading, please visit source site