Question about custom PyG and PyKeen models

Question

Question about custom PyG and PyKeen models

Alvaro-Ciudad opened this issue a year ago · comments

First of all, congratulations on a great library. I just happen to have some questions surrounding the use of custom models alongside your preprocessing module.
I have a use case that might benefit from the flexibility of implementing custom models in PyG or PyKeen, or using already implemented models for which grape might not have support already. I ve seem that embiggen has some implemented wrappers, and I wonder if you could direct me to an example of how to use those PyG and PyKeen wrappers to implement custom models.
I also think this could be a really cool tutorial to have in your documentation, and I would be ok with contributing to adding support to more PyKeen models if desired :)
Thanks

Alvaro-Ciudad · Answer 1 · Mon Jul 03 2023 18:03:31 GMT+0800 (China Standard Time)

Oh, ignore the pykeen part, I just found the wrappers for the pykeen library. Still interested in PyG, for the more complex GNNs and cuda support tho :)

Luca Cappelletti · Answer 2 · Mon Jul 03 2023 18:47:55 GMT+0800 (China Standard Time)

Hi @Alvaro-Ciudad! I still have to make that tutorial, I'll work on it later this week. The TLDR of it is to implement the abstract interface for the task that you want to make the model for.

Alvaro-Ciudad · Answer 3 · Mon Jul 03 2023 18:58:00 GMT+0800 (China Standard Time)

Great, thanks for your quick answer :)