Skip to content

semipy.datasets.get_medmnist

    semipy.datasets.cifar.get_medmnist(name: str,
                                       num_labelled: int,
                                       valid_proportion: Optional[float] = None,
                                       augmentation: bool = False,
                                       include_labelled: bool = False) -> dict

Downloads one of the MedMNIST datasets and transforms it into a Semi-Supervised dataset (with labelled and unlabelled samples).

Parameters

  • name (str) - Name of the desired dataset.
  • num_labelled (int) - Number of labelled items to include in the dataset.
  • valid_proportion (optional, float) - Size of the validation set defined by a proportion with respect to the number of labelled items. Default: None
  • augmentation (bool) - To use augmentation or not when sampling in the dataset. Default: False
  • include_labelled (bool) - To include labelled items into unlabelled set as additional data. Default: False