:py:mod:`kosmos.ml.models.model`
================================

.. py:module:: kosmos.ml.models.model







Classes
-------

.. py:class:: Model(input_dim: int, output_dim: int)

   Bases: :py:class:`torch.nn.Module`, :py:class:`abc.ABC`

   Base class for models.

   Initialize the model.

   :param input_dim: Model input dimension.
   :type input_dim: int
   :param output_dim: Model output dimension.
   :type output_dim: int





   |

   .. rubric:: Methods

   .. py:method:: forward(x: torch.Tensor) -> torch.Tensor

      Perform a forward pass.

      :param x: Input tensor.
      :type x: torch.Tensor

      :returns: Output tensor.
      :rtype: torch.Tensor