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

.. py:module:: kosmos.ml.models.neural_network







Classes
-------

.. py:class:: NeuralNetwork(input_dim: int, output_dim: int, hidden_layers: list[int])

   Bases: :py:class:`kosmos.ml.models.model.Model`

   Feedforward neural network.

   Initialize the neural network.

   :param input_dim: The input dimension of the model.
   :type input_dim: int
   :param output_dim: The output dimension of the model.
   :type output_dim: int
   :param hidden_layers: Sizes of the hidden layers. The length of the list defines
                         the number of hidden layers, and each element specifies the
                         size of a layer.
   :type hidden_layers: list[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