medcat.storage.mp_ents_save

Classes:

• BatchAnnotationSaver –

Attributes:

• logger –

logger module-attribute

logger = getLogger(__name__)

BatchAnnotationSaver

BatchAnnotationSaver(save_dir: str, batches_per_save: int)

Attributes:

• annotated_ids_path –
• batches_per_save –
• part_number –
• save_dir –

Source code in medcat-v2/medcat/storage/mp_ents_save.py

def __init__(self, save_dir: str, batches_per_save: int):
    self.save_dir = save_dir
    self.batches_per_save = batches_per_save
    self._batch_cache: list[list[
        tuple[str, Union[dict, Entities, OnlyCUIEntities]]]] = []
    os.makedirs(save_dir, exist_ok=True)
    self.part_number = 0
    self.annotated_ids_path = os.path.join(
        save_dir, "annotated_ids.pickle")

annotated_ids_path instance-attribute

annotated_ids_path = join(save_dir, 'annotated_ids.pickle')

batches_per_save instance-attribute

batches_per_save = batches_per_save

part_number instance-attribute

part_number = 0

save_dir instance-attribute

save_dir = save_dir