Implement CapsGNN

Open fanyw1997 opened this issue 5 years ago • 1 comments

CapsGNN is an interesting work outside of the paper list given in hw2. The paper URL is https://openreview.net/pdf?id=Byl8BnRcYm. A unit test is also applied.

Dec 21 '20 20:12 fanyw1997

Thanks for your contribution. I think it would be better to add a trainer for CapsGNN instead of modifying graph_classification.py in tasks. For how to add a trainer for a model, you can refer to https://github.com/THUDM/cogdl/blob/master/cogdl/trainers/sampled_trainer.py and https://github.com/THUDM/cogdl/blob/master/cogdl/models/nn/pyg_deepergcn.py.

Dec 23 '20 12:12 THINK2TRY