cellmil.datamodels.datasets.gnn_mil_dataset¶

Functions

GNNMILDataset(root, folder, label, data, ...)

cellmil.datamodels.datasets.gnn_mil_dataset.GNNMILDataset(root: Union[str, Path], folder: Union[str, Path], label: str | tuple[str, str], data: DataFrame, extractor: cellmil.interfaces.FeatureExtractorConfig.ExtractorType | list[cellmil.interfaces.FeatureExtractorConfig.ExtractorType], split: Literal['train', 'val', 'test', 'all'] = 'all', transform: Optional[Callable[[Data], Data]] = None, pre_transform: Optional[Callable[[Data], Data]] = None, pre_filter: Optional[Callable[[Data], bool]] = None, force_reload: bool = False, label_transforms: Optional[Any] = None, **kwargs: Any) → cellmil.datamodels.datasets.cell_gnn_mil_dataset.CellGNNMILDataset | cellmil.datamodels.datasets.patch_gnn_mil_dataset.PatchGNNMILDataset[source]¶