Module: tfgnn.sampler
December 14, 2023 ยท View on GitHub
View source
on GitHub
Public interface for GNN Sampler.
Classes
class SamplingOp: A ProtocolMessage
class SamplingSpec: A ProtocolMessage
class SamplingSpecBuilder: Mimics
builder pattern that eases creation of tfgnn.SamplingSpec.
Functions
make_sampling_spec_tree(...):
Automatically creates SamplingSpec by starting from seed node set.
Other Members | |
|---|---|
| SamplingStrategy | ['TOP_K', 'RANDOM_UNIFORM', 'RANDOM_WEIGHTED'] |