Categories
English Machine Learning

Short comparison between graph neural network libraries: Graph nets VS Stellagraph VS Spektral

 

FeatureSpektralStellargraphGraph Nets
Tutorials availabilityThere’re about 20 examples that show different usages.A lot of well documented demos in python notebooks.Very few examples
Year of creation202020182018
PaperSpectral Clustering with Graph Neural Networks for Graph Pooling cited less than 100 timesNo paperRelational inductive biases, deep learning, and graph networks cited more than 1400 times
Author/sdanielegrattarola

 

CSIRO’s Data61DeepMind
SupportsTensorflowTensorflowTensorflow and Sonnet
Input typeSimple datasets of multiple thousands single numerical value nodesDataframesIn the examples the node values were required to be arrays of float values. It’s also possible to choose whether to make a graph without edges, nodes without values, or edges without directions.
UsageExamples show a case of citations network, molecular prediction model, and graph network classification.Varied use cases of supervised and unsupervised learning. It also supports 20 different types of algorithms.Examples show one physical example, one about the shortest path problem solution and another simplified demo.
Node predictionYesClassificationDoesn’t predict graphs, nodes or edges according to the manual, but predicts states, shortest paths and other operations that are done over an existing graph.
Edge predictionNot in the examplesYes (link prediction)
Graph predictionYesClassification
Support categorical dataYes, in one hot vector.

 

Leave a Reply

Your email address will not be published. Required fields are marked *