:py:mod:`ggt.data`
==================

.. py:module:: ggt.data


Submodules
----------
.. toctree::
   :titlesonly:
   :maxdepth: 1

   dataset/index.rst
   make_splits/index.rst


Package Contents
----------------

Classes
~~~~~~~

.. autoapisummary::

   ggt.data.FITSDataset



Functions
~~~~~~~~~

.. autoapisummary::

   ggt.data.get_data_loader



.. py:class:: FITSDataset(data_dir, slug=None, split=None, channels=1, cutout_size=167, label_col='bt_g', normalize=True, transform=None, expand_factor=1, repeat_dims=False, label_scaling=None, scaling_data_dir=None, scaling_slug=None, load_labels=True)

   Bases: :py:obj:`torch.utils.data.Dataset`

   Dataset from FITS files. Pre-caches FITS files as PyTorch tensors to
   improve data load speed.

   .. py:method:: __getitem__(index)

      Magic method to index into the dataset.


   .. py:method:: __len__()

      Return the effective length of the dataset.


   .. py:method:: load_fits_as_tensor(filename)
      :staticmethod:

      Open a FITS file and convert it to a Torch tensor.



.. py:function:: get_data_loader(dataset, batch_size, n_workers, shuffle=True)