mllf.cb.rgcn

RGCN encoder using PyTorch Geometric.

Implements a small multi-layer RGCN that returns node embeddings of fixed dimension. Requires torch and torch_geometric to be installed in the runtime where training occurs.

Classes

RGCNEncoder(in_dim, hidden_dims, out_dim, ...)