emmi.callbacks.supernode_pooling_coverage

Classes

SupernodePoolingCoverageCallback

Hooks into the message MLP of the supernode pooling to retrieve the coverage/degree of the radius graph.

Module Contents

class emmi.callbacks.supernode_pooling_coverage.SupernodePoolingCoverageCallback(supernode_pooling_path='encoder', **kwargs)

Bases: ksuit.callbacks.PeriodicCallback

Hooks into the message MLP of the supernode pooling to retrieve the coverage/degree of the radius graph.

Initializes the PeriodicCallback.

Parameters:

  • callback_config – Configuration of the PeriodicCallback. Implements the CallBackBaseConfig schema.

  • trainer – Trainer of the current run, subclass of SgdTrainer.

  • model – Model of the current run.

  • data_container – DataContainer instance that provides access to all datasets.

  • tracker – Tracker instance to log metrics to stdout/disk/online platform.

  • log_writer – LogWriter instance to log metrics.

  • checkpoint_writer – CheckpointWriter instance to save checkpoints.

  • metric_property_provider – MetricPropertyProvider instance to access properties of metrics.

  • name – Name of the callback.

supernode_pooling_path = 'encoder'
hook_outputs
pos_embed_hook
dst_idx_hook
src_idx_hook
tracked_degrees = []
tracked_coverages = []