`medcat.utils.relation_extraction.bert.config`

Module Contents

Classes

BertConfig_RelationExtraction

Class for BertConfig

Attributes

medcat.utils.relation_extraction.bert.config.logger

class medcat.utils.relation_extraction.bert.config.BertConfig_RelationExtraction(pretrained_model_name_or_path, **kwargs)

Bases: medcat.utils.relation_extraction.config.BaseConfig_RelationExtraction

Class for BertConfig

name = 'bert-config'

pretrained_model_name_or_path = 'bert-base-uncased'

hf_model_config: transformers.BertConfig

classmethod load(pretrained_model_name_or_path, relcat_config, **kwargs)

Parameters:

pretrained_model_name_or_path (str) –
relcat_config (medcat.config_rel_cat.ConfigRelCAT) –

Return type:

BertConfig_RelationExtraction