Index

_

__call__() (haiku_geometric.models.MLP method) (haiku_geometric.models.Node2Vec method) (haiku_geometric.nn.attention.SelfAttention method) (haiku_geometric.nn.attention.Transformer method) (haiku_geometric.nn.conv.EdgeConv method) (haiku_geometric.nn.conv.GATConv method) (haiku_geometric.nn.conv.GatedGraphConv method) (haiku_geometric.nn.conv.GCNConv method) (haiku_geometric.nn.conv.GeneralConv method) (haiku_geometric.nn.conv.GINConv method) (haiku_geometric.nn.conv.GINEConv method) (haiku_geometric.nn.conv.GPSLayer method) (haiku_geometric.nn.conv.GraphConv method) (haiku_geometric.nn.conv.MetaLayer method) (haiku_geometric.nn.conv.PNAConv method) (haiku_geometric.nn.conv.SAGEConv method) (haiku_geometric.nn.dense.Linear method) (haiku_geometric.nn.pool.TopKPooling method) (haiku_geometric.posenc.LaplacianEncoder method) (haiku_geometric.posenc.MagLaplacianEncoder method)

__init__() (haiku_geometric.models.MLP method) (haiku_geometric.models.Node2Vec method) (haiku_geometric.nn.attention.Transformer method) (haiku_geometric.nn.conv.EdgeConv method) (haiku_geometric.nn.conv.GATConv method) (haiku_geometric.nn.conv.GatedGraphConv method) (haiku_geometric.nn.conv.GCNConv method) (haiku_geometric.nn.conv.GeneralConv method) (haiku_geometric.nn.conv.GINConv method) (haiku_geometric.nn.conv.GINEConv method) (haiku_geometric.nn.conv.GPSLayer method) (haiku_geometric.nn.conv.GraphConv method) (haiku_geometric.nn.conv.MetaLayer method) (haiku_geometric.nn.conv.PNAConv method) (haiku_geometric.nn.conv.SAGEConv method) (haiku_geometric.nn.dense.Linear method) (haiku_geometric.nn.pool.TopKPooling method) (haiku_geometric.posenc.LaplacianEncoder method) (haiku_geometric.posenc.MagLaplacianEncoder method)

A

add_self_loops() (in module haiku_geometric.transforms)

Aggregation (class in haiku_geometric.nn.aggr)

B

batch() (in module haiku_geometric.utils)

C

coalesce() (in module haiku_geometric.utils)

D

DataGraphTuple (class in haiku_geometric.datasets)

DegreeScalerAggregation (class in haiku_geometric.nn.aggr)

E

EdgeConv (class in haiku_geometric.nn.conv) edges (haiku_geometric.datasets.DataGraphTuple attribute)

eigv_laplacian() (in module haiku_geometric.utils) eigv_magnetic_laplacian() (in module haiku_geometric.utils)

F

fill_diagonal() (in module haiku_geometric.utils)

G

GATConv (class in haiku_geometric.nn.conv) GatedGraphConv (class in haiku_geometric.nn.conv) GCNConv (class in haiku_geometric.nn.conv) GeneralConv (class in haiku_geometric.nn.conv) get_laplacian() (in module haiku_geometric.utils) get_laplacian_matrix() (in module haiku_geometric.utils) GINConv (class in haiku_geometric.nn.conv) GINEConv (class in haiku_geometric.nn.conv)

global_add_pool() (in module haiku_geometric.nn.pool) global_max_pool() (in module haiku_geometric.nn.pool) global_mean_pool() (in module haiku_geometric.nn.pool) globals (haiku_geometric.datasets.DataGraphTuple attribute) GNNBenchmarkDataset (class in haiku_geometric.datasets) GPSLayer (class in haiku_geometric.nn.conv) GraphConv (class in haiku_geometric.nn.conv) GraphDataset (class in haiku_geometric.datasets)

H

haiku_geometric.datasets module haiku_geometric.models module haiku_geometric.nn.aggr module haiku_geometric.nn.attention module haiku_geometric.nn.conv module

haiku_geometric.nn.dense module haiku_geometric.nn.pool module haiku_geometric.posenc module haiku_geometric.transforms module haiku_geometric.utils module

K

KarateClub (class in haiku_geometric.datasets)

L

LaplacianEncoder (class in haiku_geometric.posenc)

lift() (haiku_geometric.nn.conv.GATConv method) Linear (class in haiku_geometric.nn.dense)

M

MagLaplacianEncoder (class in haiku_geometric.posenc) MaxAggregation (class in haiku_geometric.nn.aggr) MeanAggregation (class in haiku_geometric.nn.aggr) MetaLayer (class in haiku_geometric.nn.conv) MinAggregation (class in haiku_geometric.nn.aggr) MLP (class in haiku_geometric.models) module haiku_geometric.datasets haiku_geometric.models haiku_geometric.nn.aggr haiku_geometric.nn.attention haiku_geometric.nn.conv haiku_geometric.nn.dense haiku_geometric.nn.pool haiku_geometric.posenc haiku_geometric.transforms haiku_geometric.utils

MultiAggregation (class in haiku_geometric.nn.aggr)

N

n_edge (haiku_geometric.datasets.DataGraphTuple attribute) n_node (haiku_geometric.datasets.DataGraphTuple attribute) neg_sample() (haiku_geometric.models.Node2Vec method)

Node2Vec (class in haiku_geometric.models) nodes (haiku_geometric.datasets.DataGraphTuple attribute) normalize_features() (in module haiku_geometric.transforms)

O

OGB (class in haiku_geometric.datasets)

P

pad_graph() (in module haiku_geometric.utils) Planetoid (class in haiku_geometric.datasets)

PNAConv (class in haiku_geometric.nn.conv) pos_sample() (haiku_geometric.models.Node2Vec method) position (haiku_geometric.datasets.DataGraphTuple attribute)

R

random_walk() (in module haiku_geometric.utils)

receivers (haiku_geometric.datasets.DataGraphTuple attribute) remove_self_loops() (in module haiku_geometric.transforms)

S

SAGEConv (class in haiku_geometric.nn.conv) sample() (haiku_geometric.models.Node2Vec method) scatter() (in module haiku_geometric.utils)

SelfAttention (class in haiku_geometric.nn.attention) senders (haiku_geometric.datasets.DataGraphTuple attribute) SumAggregation (class in haiku_geometric.nn.aggr)

T

to_undirected() (in module haiku_geometric.utils) TopKPooling (class in haiku_geometric.nn.pool) ToyGraphDataset (class in haiku_geometric.datasets)

train_mask (haiku_geometric.datasets.DataGraphTuple attribute) Transformer (class in haiku_geometric.nn.attention) TUDataset (class in haiku_geometric.datasets)

U

unbatch() (in module haiku_geometric.utils)

Y

y (haiku_geometric.datasets.DataGraphTuple attribute)