cogdl.data.sampler
¶
Module Contents¶
Classes¶
-
class
cogdl.data.sampler.
SAINTSampler
(data, args_params)[source]¶ Bases:
cogdl.data.sampler.Sampler
-
get_subgraph
(self, phase, require_norm=True)[source]¶ Generate one minibatch for model. In the ‘train’ mode, one minibatch corresponds to one subgraph of the training graph. In the ‘valid’ or ‘test’ mode, one batch corresponds to the full graph (i.e., full-batch rather than minibatch evaluation for validation / test sets).
- Inputs:
mode str, can be ‘train’, ‘valid’, ‘test’ require_norm boolean
- Outputs:
data Data object, modeling the sampled subgraph data.norm_aggr aggregation normalization data.norm_loss normalization normalization
-
-
class
cogdl.data.sampler.
LayerSampler
(data, model, params_args)[source]¶ Bases:
cogdl.data.sampler.Sampler