medcat.utils.ner.data_collator

Classes

CollateAndPadNER

Module Contents

class medcat.utils.ner.data_collator.CollateAndPadNER(pad_id)

Bases: object

__init__(pad_id)
pad_id
__call__(features)
Parameters:

features (List[Any])

Return type:

Dict[str, torch.Tensor]