Namespace AiDotNet.NeuralNetworks.Tasks.Graph

Classes

GraphClassificationModel<T>

Implements a complete neural network model for graph classification tasks.

LinkPredictionModel<T>

Implements a complete neural network model for link prediction tasks on graphs.

NodeClassificationModel<T>

Implements a complete neural network model for node classification tasks on graphs.

Enums

GraphClassificationModel<T>.GraphPooling

Graph pooling methods for aggregating node embeddings.

LinkPredictionModel<T>.LinkPredictionDecoder

Decoder types for combining node embeddings into edge scores.