cellmil.datamodels.datasets.mil_dataset

Functions

MILDataset(root, label, folder, data, extractor)

cellmil.datamodels.datasets.mil_dataset.MILDataset(root: Path, label: str | tuple[str, str], folder: Path, data: DataFrame, extractor: cellmil.interfaces.FeatureExtractorConfig.ExtractorType | list[cellmil.interfaces.FeatureExtractorConfig.ExtractorType], split: Literal['train', 'val', 'test', 'all'] = 'all', **kwargs: Any) → cellmil.datamodels.datasets.cell_mil_dataset.CellMILDataset | cellmil.datamodels.datasets.patch_mil_dataset.PatchMILDataset