train_tools.data_utils.datasetter
=================================

.. py:module:: train_tools.data_utils.datasetter




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

.. py:function:: get_dataloaders_labeled(root, mapping_file, mapping_file_tuning, join_mapping_file=None, valid_portion=0.0, batch_size=8, amplified=False, relabel=False)

   Set DataLoaders for labeled datasets.

   :param root: root directory
   :type root: str
   :param mapping_file: json file for mapping dataset
   :type mapping_file: str
   :param valid_portion: portion of valid datasets. Defaults to 0.1.
   :type valid_portion: float, optional
   :param batch_size: batch size. Defaults to 8.
   :type batch_size: int, optional
   :param shuffle: shuffles dataloader. Defaults to True.
   :type shuffle: bool, optional
   :param num_workers: number of workers for each datalaoder. Defaults to 5.
   :type num_workers: int, optional

   :returns: dictionary of data loaders.
   :rtype: dict


.. py:function:: get_dataloaders_public(root, mapping_file, valid_portion=0.0, batch_size=8)

   Set DataLoaders for labeled datasets.

   :param root: root directory
   :type root: str
   :param mapping_file: json file for mapping dataset
   :type mapping_file: str
   :param valid_portion: portion of valid datasets. Defaults to 0.1.
   :type valid_portion: float, optional
   :param batch_size: batch size. Defaults to 8.
   :type batch_size: int, optional
   :param shuffle: shuffles dataloader. Defaults to True.
   :type shuffle: bool, optional

   :returns: dictionary of data loaders.
   :rtype: dict


.. py:function:: get_dataloaders_unlabeled(root, mapping_file, batch_size=8, shuffle=True, num_workers=5)

   Set dataloaders for unlabeled dataset.