djl icon indicating copy to clipboard operation
djl copied to clipboard

Does DJL have a gnn?

Open SidneyLann opened this issue 2 years ago • 6 comments

Does DJL have a gnn?

SidneyLann avatar Feb 28 '22 13:02 SidneyLann

If you mean something like a GAN, then there is an example: https://docs.djl.ai/examples/docs/biggan.html

lanking520 avatar Feb 28 '22 18:02 lanking520

Like tf-gnn

SidneyLann avatar Feb 28 '22 21:02 SidneyLann

We don't currently have helpers for graph neural networks, but you should be able to implement them assuming the base engine supports them. Are you able to share your use case for gnns?

zachgk avatar Mar 01 '22 21:03 zachgk

I need a tf-gat like gnn in java, thanks.

SidneyLann avatar Mar 01 '22 23:03 SidneyLann

https://github.com/tensorflow/gnn/blob/main/tensorflow_gnn/graph/keras/layers/gat_v2.py

SidneyLann avatar Mar 02 '22 00:03 SidneyLann

Or all other gnn in java.

SidneyLann avatar Mar 06 '22 14:03 SidneyLann