:py:mod:`kosmos.ml.datasets.dataset`
====================================

.. py:module:: kosmos.ml.datasets.dataset







Classes
-------

.. py:class:: SLDataset(x: collections.abc.Collection, y: collections.abc.Collection, *, min_max_scaler: bool = True)

   Bases: :py:class:`torch.utils.data.Dataset[tuple[torch.Tensor, torch.Tensor]]`, :py:class:`abc.ABC`

   Dataset for supervised learning classification.

   Initialize the dataset.

   :param x: Feature values.
   :type x: Collection
   :param y: Target values.
   :type y: Collection
   :param min_max_scaler: Whether to apply min-max scaling to the features.
   :type min_max_scaler: bool




   |

   .. rubric:: Properties

   .. py:property:: input_dimension

      Number of feature columns.

   .. py:property:: output_dim

      Number of distinct classes.

   .. py:property:: class_names

      Class names.