Skip to content

semipy.datasets.get_dataset

Warning

This section is in construction.

    semipy.datasets.get_dataset(method_name: str,
                                data_config: Dict) -> Dict[str, torch.utils.data.Dataset]

This function allows to call the right dataset getter function depending in the asked dataset. It returns a dictionary containing at least the training set, and also the validation and test set if those are asked by the user or already present in the original dataset.

Parameters

  • method_name (str) - Name of the used SSL method (e.g 'fixmatch').
  • data_config (dict) - Dictionary of parameters dedicated to data (DATA part of the configuration file).