Enum GraphClassificationModel<T>.GraphPooling

Namespace

AiDotNet.NeuralNetworks.Tasks.Graph

Assembly

AiDotNet.dll

Graph pooling methods for aggregating node embeddings.

public enum GraphClassificationModel<T>.GraphPooling

Fields

Attention = 3

Attention pooling: Weighted average with learned attention.

Max = 1

Max pooling: Take max across all node embeddings.

Mean = 0

Mean pooling: Average all node embeddings.

Sum = 2

Sum pooling: Sum all node embeddings.