hyrax.datasets.hyrax_cifar_dataset
==================================

.. py:module:: hyrax.datasets.hyrax_cifar_dataset


Attributes
----------

.. autoapisummary::

   hyrax.datasets.hyrax_cifar_dataset.logger


Classes
-------

.. autoapisummary::

   hyrax.datasets.hyrax_cifar_dataset.HyraxCifarDataset


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

.. py:data:: logger

.. py:class:: HyraxCifarDataset(config: dict, data_location: pathlib.Path = None)

   Bases: :py:obj:`hyrax.datasets.dataset_registry.HyraxDataset`


   Map style CIFAR 10 dataset for Hyrax

   This utilizes the CIFAR dataset from torchvision for retrieving the dataset.

   .. py:method:: __init__

   Overall initialization for all Datasets which saves the config

   Subclasses of HyraxDataset ought call this at the end of their __init__ like:

   .. code-block:: python

       from hyrax.datasets import HyraxDataset

       class MyDataset(HyraxDataset):
           def __init__(config):
               <your code>
               super().__init__(config)

   If per tensor metadata is available, it is recommended that dataset authors create an
   astropy Table of that data, in the same order as their data and pass that `metadata_table`
   as shown below:

   .. code-block:: python

       from hyrax.datasets import HyraxDataset
       from astropy.table import Table

       class MyDataset(HyraxDataset):
           def __init__(config):
               <your code>
               metadata_table = Table(<Your catalog data goes here>)
               super().__init__(config, metadata_table)

   :param config: The runtime configuration for hyrax
   :type config: dict, Optional
   :param metadata_table: An Astropy Table with
                          1. the metadata columns desired for visualization AND
                          2. in the order your data will be enumerated.
   :type metadata_table: Optional[Table], optional
   :param object_id_column_name: The name of the column containing object IDs. If None, uses the default
                                 from config or creates one from the ids() method.
   :type object_id_column_name: Optional[str], optional


   .. py:attribute:: data_location
      :value: None



   .. py:attribute:: training_data


   .. py:attribute:: cifar


   .. py:attribute:: id_width
      :value: 0



   .. py:method:: get_image(idx)

      Get the image at the given index as a NumPy array.



   .. py:method:: get_label(idx)

      Get the label at the given index.



   .. py:method:: get_object_id(idx)

      Get the object ID for the item as a string.



   .. py:method:: __len__()


