Machine Learning ml

Machine Learning ml#

ml.MILDataset(dd[, donor_layer, ...])

PyTorch Dataset for Multiple Instance Learning (MIL) using donor- and cell-level data from AnnData or MuData objects.

ml.mil_collate_fn(batch)

Custom collate function for MILDataset to prepare batched input for PyTorch models.

ml.DonorMILModel([n_input_donor, ...])

A PyTorch Lightning model for multiple instance learning (MIL) using donor-level and cell-level features.