How to train my own dataset

iMoonLab / HGNN

Hypergraph Neural Networks (AAAI 2019)

MIT License

661 stars 128 forks source link

Closed liuyaaaaang closed 5 years ago

liuyaaaaang commented 5 years ago

Excuse me,could I train my own datasets on HGNN?

yifanfeng97 commented 5 years ago

Sure. You can design function to preprocess your dataset. Then node features and hypergraph incidence matrix are fed into the model.