medcat.utils.relation_extraction.config

Module Contents

Classes

BaseConfig_RelationExtraction

Base class for the RelCAT models

Attributes

logger

medcat.utils.relation_extraction.config.logger
class medcat.utils.relation_extraction.config.BaseConfig_RelationExtraction(pretrained_model_name_or_path, **kwargs)

Bases: transformers.PretrainedConfig

Base class for the RelCAT models

name = 'base-config-relcat'
__init__(pretrained_model_name_or_path, **kwargs)
to_dict()

Serializes this instance to a Python dictionary.

Returns:

`Dict[str, Any]` – Dictionary of all the attributes that make up this configuration instance.

save(save_path)
Parameters:

save_path (str) –

classmethod load(pretrained_model_name_or_path, relcat_config, **kwargs)
Parameters:
Return type:

BaseConfig_RelationExtraction