medcat.datasets.data_collator

Module Contents

Classes

CollateAndPadNER

class medcat.datasets.data_collator.CollateAndPadNER(pad_id)

Bases: object

__init__(pad_id)

__call__(features)

Parameters:

features (List[Any]) –

Return type:

Dict[str, torch.Tensor]