search

deepjavalibrary / djl

An Engine-Agnostic Deep Learning Framework in Java
https://djl.ai
Apache License 2.0
4.05k stars 648 forks source link

Does DJL have a gnn? #1514

Open SidneyLann opened 2 years ago

SidneyLann commented 2 years ago

Does DJL have a gnn?

lanking520 commented 2 years ago

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

SidneyLann commented 2 years ago

Like tf-gnn

zachgk commented 2 years ago

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?

SidneyLann commented 2 years ago

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

SidneyLann commented 2 years ago

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

SidneyLann commented 2 years ago

Or all other gnn in java.