cellink.ml.mil_collate_fn#
- cellink.ml.mil_collate_fn(batch)#
Custom collate function for MILDataset to prepare batched input for PyTorch models.
Donor-level features are stacked into tensors, while cell-level features remain as lists of tensors corresponding to each donor. This preserves the MIL structure.
- Parameters:
batch (list[dict]) – List of samples returned by MILDataset.__getitem__.
- Returns:
dict Batched sample with stacked donor-level tensors and list-based cell-level tensors. Keys include: - ‘donor_x’, ‘donor_y’, ‘donor_batch’, ‘donor_cat_covs’, ‘donor_cont_covs’, ‘donor_indices’ - ‘cell_x’, ‘cell_y’, ‘cell_batch’, ‘cell_cat_covs’, ‘cell_cont_covs’, ‘cell_indices’