View source
on GitHub
Public interface for GNN Sampler.
class SamplingOp: A ProtocolMessage
class SamplingSpec: A ProtocolMessage
class SamplingSpecBuilder: Mimics
builder pattern that eases creation of tfgnn.SamplingSpec.
make_sampling_spec_tree(...):
Automatically creates SamplingSpec by starting from seed node set.
| SamplingStrategy | ['TOP_K', 'RANDOM_UNIFORM', 'RANDOM_WEIGHTED'] |