emmi.callbacks.supernode_pooling_coverage
=========================================

.. py:module:: emmi.callbacks.supernode_pooling_coverage


Classes
-------

.. autoapisummary::

   emmi.callbacks.supernode_pooling_coverage.SupernodePoolingCoverageCallback


Module Contents
---------------

.. py:class:: SupernodePoolingCoverageCallback(supernode_pooling_path='encoder', **kwargs)

   Bases: :py:obj:`ksuit.callbacks.PeriodicCallback`


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

   Initializes the `PeriodicCallback`.

   :param callback_config: Configuration of the `PeriodicCallback`. Implements the `CallBackBaseConfig` schema.
   :param trainer: Trainer of the current run, subclass of `SgdTrainer`.
   :param model: Model of the current run.
   :param data_container: DataContainer instance that provides access to all datasets.
   :param tracker: Tracker instance to log metrics to stdout/disk/online platform.
   :param log_writer: LogWriter instance to log metrics.
   :param checkpoint_writer: CheckpointWriter instance to save checkpoints.
   :param metric_property_provider: MetricPropertyProvider instance to access properties of metrics.
   :param name: Name of the callback.


   .. py:attribute:: supernode_pooling_path
      :value: 'encoder'



   .. py:attribute:: hook_outputs


   .. py:attribute:: pos_embed_hook


   .. py:attribute:: dst_idx_hook


   .. py:attribute:: src_idx_hook


   .. py:attribute:: tracked_degrees
      :value: []



   .. py:attribute:: tracked_coverages
      :value: []