gnn_tracking.models.noise_classification
========================================

.. py:module:: gnn_tracking.models.noise_classification

.. autoapi-nested-parse::

   Models for filtering out noise before we even build a graph



Classes
-------

.. autoapisummary::

   gnn_tracking.models.noise_classification.TruthNoiseClassifierModel
   gnn_tracking.models.noise_classification.WithNoiseClassification


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

.. py:class:: TruthNoiseClassifierModel

   Bases: :py:obj:`torch.nn.Module`


   Remove all noise with truth information


   .. py:method:: forward(data: torch_geometric.data.Data) -> torch_geometric.data.Data


.. py:class:: WithNoiseClassification(noise_model: torch.nn.Module, model: torch.nn.Module)

   Bases: :py:obj:`torch.nn.Module`, :py:obj:`pytorch_lightning.core.mixins.hparams_mixin.HyperparametersMixin`


   Combine a noise filter with another model


   .. py:attribute:: noise_model


   .. py:attribute:: model


   .. py:method:: forward(data: torch_geometric.data.Data) -> dict[str, torch.Tensor | None]