medcat.components.addons.relation_extraction.rel_cat

Attributes

`COMPONENTS_FOLDER`
`logger`

Classes

`CDB`	The abstract serialisable base class.
`Vocab`	Vocabulary used to store word embeddings for context similarity
`Config`	The base serialisable config.
`ComponentConfig`	The base serialisable config.
`ConfigRelCAT`	The RelCAT part of the config
`SerialisingStrategy`	Describes the strategy for serialising.
`AddonComponent`	Base/abstract addon component class.
`RelExtrBaseComponent`
`RelData`	An abstract class representing a `Dataset`.
`BaseTokenizer`	The base tokenizer protocol.
`MutableDocument`	The mutable parts of the document.
`RelCATAddon`	Base/abstract addon component class.
`BalancedBatchSampler`	Base class for all Samplers.
`RelCAT`	The RelCAT class used for training 'Relation-Annotation' models, i.e.,

Functions

`deserialise`(folder_path[, ignore_folders_prefix, ...])	Deserialise contents of a folder.
`set_all_seeds`(seed)
`load_results`(path[, model_name, file_prefix])
`load_state`(model, optimizer, scheduler[, path, ...])	Used by RelCAT.load() and RelCAT.train()
`save_results`(data[, model_name, path, file_prefix])
`save_state`(model, optimizer, scheduler[, epoch, ...])	Used by RelCAT.save() and RelCAT.train()
`split_list_train_test_by_class`(data[, sample_limit, ...])
`create_tokenizer`(tokenizer_name, config)	Create the tokenizer given the init arguments.

Module Contents

class medcat.components.addons.relation_extraction.rel_cat.CDB(config)

Bases: medcat.storage.serialisables.AbstractSerialisable

The abstract serialisable base class.

This defines some common defaults.

Parameters:: config (medcat.config.Config)

__init__(config)

Parameters:: config (medcat.config.Config)
Return type:: None

config

cui2info: dict[str, medcat.cdb.concepts.CUIInfo]

name2info: dict[str, medcat.cdb.concepts.NameInfo]

type_id2info: dict[str, medcat.cdb.concepts.TypeInfo]

token_counts: dict[str, int]

addl_info: dict[str, Any]

_subnames: set[str]

is_dirty = False

has_changed_names = False

classmethod get_init_attrs()

Return type:: list[str]

_reset_subnames()

has_subname(name)

Whether the CDB has the specified subname.

Parameters:: name (str) – The subname to check.
Returns:: bool – Whether the subname is present in this CDB.
Return type:: bool

get_name(cui)

Returns preferred name if it exists, otherwise it will return the longest name assigned to the concept.

Parameters:: cui (str) – Concept ID or unique identifier in this database.
Returns:: str – The name of the concept.
Return type:: str

weighted_average_function(step)

Get the weighted average for steop.

Parameters:: step (int) – The steop.
Returns:: float – The weighted average.
Return type:: float

add_types(types)

Add type info to CDB.

Parameters:: types (Iterable[tuple[str, str]]) – The raw type info.
Return type:: None

add_names(cui, names, name_status=ST.AUTOMATIC, full_build=False)

Adds a name to an existing concept.

Parameters:

cui (str) – Concept ID or unique identifier in this database, all concepts that have the same CUI will be merged internally.
names (dict[str, NameDescriptor]) –
Names for this concept, or the value that if found in free text can be linked to this concept. Names is an dict like: `{name: {‘tokens’: tokens, ‘snames’: snames,

’raw_name’: raw_name}, …}`

Names should be generated by helper function ‘medcat.preprocessing.cleaners.prepare_name’
name_status (str) – One of P, N, A. Defaults to ‘A’.
full_build (bool) – If True the dictionary self.addl_info will also be populated, contains a lot of extra information about concepts, but can be very memory consuming. This is not necessary for normal functioning of MedCAT (Default value False).

Return type:

None

_add_concept_names(cui, names, name_status)

Parameters:

cui (str)
names (dict[str, medcat.preprocessors.cleaners.NameDescriptor])
name_status (str)

Return type:

None

_add_full_build(cui, names, ontologies, description, type_ids)

Parameters:

cui (str)
names (dict[str, medcat.preprocessors.cleaners.NameDescriptor])
ontologies (set[str])
description (str)
type_ids (set[str])

Return type:

None

_add_concept(cui, names, ontologies, name_status, type_ids, description, full_build=False)

Add a concept to internal Concept Database (CDB). Depending on what you are providing this will add a large number of properties for each concept.

Parameters:

cui (str) – Concept ID or unique identifier in this database, all concepts that have the same CUI will be merged internally.
names (dict[str, NameDescriptor]) –
Names for this concept, or the value that if found in free text can be linked to this concept. Names is a dict like: `{name: {‘tokens’: tokens, ‘snames’: snames,

’raw_name’: raw_name}, …}`

Names should be generated by helper function ‘medcat.preprocessing.cleaners.prepare_name’
ontologies (set[str]) – ontologies in which the concept exists (e.g. SNOMEDCT, HPO)
name_status (str) – One of P, N, A
type_ids (set[str]) – Semantic type identifier (have a look at TUIs in UMLS or SNOMED-CT)
description (str) – Description of this concept.
full_build (bool) – If True the dictionary self.addl_info will also be populated, contains a lot of extra information about concepts, but can be very memory consuming. This is not necessary for normal functioning of MedCAT (Default Value False).

Return type:

None

reset_training()

Will remove all training efforts - in other words all embeddings that are learnt for concepts in the current CDB. Please note that this does not remove synonyms (names) that were potentially added during supervised/online learning.

Return type:: None

filter_by_cui(cuis_to_keep)

Subset the core CDB fields (dictionaries/maps).

Note that this will potenitally keep a bit more CUIs then in cuis_to_keep. It will first find all names that link to the cuis_to_keep and then find all CUIs that link to those names and keep all of them.

This also will not remove any data from cdb.addl_info - as this field can contain data of unknown structure.

Parameters:: cuis_to_keep (Collection[str]) – CUIs that will be kept, the rest will be removed (not completely, look above).
Raises:: Exception – If no snames and subsetting is not possible.
Return type:: None

remove_cui(cui)

This function takes a CUI and removes it the CDB.

It also removes the CUI from name specific per_cui_status maps as well as well as removes all the names that do not correspond to any CUIs after the removal of this one.

Parameters:: cui (str) – The CUI to remove.
Return type:: None

_remove_names(cui, names)

Remove names from an existing concept - effect is this name will never again be used to link to this concept. This will only remove the name from the linker (namely name2cuis and name2cuis2status), the name will still be present everywhere else. Why? Because it is bothersome to remove it from everywhere, but could also be useful to keep the removed names in e.g. cui2names.

Parameters:

cui (str) – Concept ID or unique identifier in this database.
names (Iterable[str]) – Names to be removed (e.g list, set, or even a dict (in which case keys will be used)).

Return type:

None

__eq__(other)

Parameters:: other (Any)
Return type:: bool

get_cui2count_train()

Return type:: dict[str, int]

get_name2count_train()

Return type:: dict[str, int]

get_hash()

Return type:: str

get_basic_info()

Return type:: medcat.data.model_card.CDBInfo

save(save_path, serialiser=AvailableSerialisers.dill, overwrite=False)

Save CDB at path.

Parameters:

save_path (str) – The path to save at.
serialiser (Union[ str, AvailableSerialisers], optional) – The serialiser. Defaults to AvailableSerialisers.dill.
overwrite (bool, optional) – Whether to allow overwriting existing files. Defaults to False.

Return type:

None

classmethod load(path)

Parameters:: path (str)
Return type:: CDB

get_strategy()

Return type:: SerialisingStrategy

classmethod ignore_attrs()

Return type:: list[str]

classmethod include_properties()

Return type:: list[str]

class medcat.components.addons.relation_extraction.rel_cat.Vocab

Bases: medcat.storage.serialisables.AbstractSerialisable

Vocabulary used to store word embeddings for context similarity calculation. Also used by the spell checker - but not for fixing the spelling only for checking is something correct.

Properties:

vocab (dict[str, WordDescriptor]):

Map from word to attributes, e.g. {‘house’:: {‘vector’: <np.array>, ‘count’: <int>, …}, …}

index2word (dict[int, str]):

From word to an index - used for negative sampling

vec_index2word (dict):

Same as index2word but only words that have vectors

__init__()

Return type:: None

vocab: dict[str, WordDescriptor]

index2word: dict[int, str]

vec_index2word: dict[int, str]

cum_probs: numpy.ndarray

inc_or_add(word, cnt=1, vec=None)

Add a word or increase its count.

Parameters:

word (str) – Word to be added
cnt (int) – By how much should the count be increased, or to what should it be set if a new word. (Default value = 1)
vec (Optional[np.ndarray]) – Word vector (Default value = None)

Return type:

None

remove_all_vectors()

Remove all stored vector representations.

Return type:: None

remove_words_below_cnt(cnt)

Remove all words with frequency below cnt.

Parameters:: cnt (int) – Word count limit.
Return type:: None

_rebuild_index()

inc_wc(word, cnt=1)

Incraese word count by cnt.

Parameters:

word (str) – For which word to increase the count
cnt (int) – By how muhc to increase the count (Default value = 1)

Return type:

None

add_vec(word, vec)

Add vector to a word.

Parameters:

word (str) – To which word to add the vector.
vec (np.ndarray) – The vector to add.

Return type:

None

reset_counts(cnt=1)

Reset the count for all word to cnt.

Parameters:: cnt (int) – New count for all words in the vocab. (Default value = 1)
Return type:: None

update_counts(tokens)

Given a list of tokens update counts for words in the vocab.

Parameters:: tokens (list[str]) – Usually a large block of text split into tokens/words.
Return type:: None

add_word(word, cnt=1, vec=None, replace=True)

Add a word to the vocabulary

Parameters:

word (str) – The word to be added, it should be lemmatized and lowercased
cnt (int) – Count of this word in your dataset (Default value = 1)
vec (Optional[np.ndarray]) – The vector representation of the word (Default value = None)
replace (bool) – Will replace old vector representation (Default value = True)

Return type:

None

add_words(path, replace=True)

Adds words to the vocab from a file, the file is required to have the following format (vec being optional):

<word> <cnt>[ <vec_space_separated>]

e.g. one line: the word house with 3 dimensional vectors: house 34444 0.3232 0.123213 1.231231

Parameters:

path (str) – path to the file with words and vectors
replace (bool) – existing words in the vocabulary will be replaced. Defaults to True.

Return type:

None

init_cumsums()

Initialise cumulative sums.

This is in place of the unigram table. But similarly to it, this approach allows generating a list of indices that match the probabilistic distribution expected as per the word counts of each word.

Return type:: None

get_negative_samples(n=6, ignore_punct_and_num=False)

Get N negative samples.

Parameters:

n (int) – How many words to return (Default value = 6)
ignore_punct_and_num (bool) – Whether to ignore punctuation and numbers. Defaults to False.

Raises:

Exception – If no unigram table is present.

Returns:

list[int] – Indices for words in this vocabulary.

Return type:

list[int]

get_vectors(indices)

Parameters:: indices (list[int])
Return type:: list[numpy.ndarray]

__getitem__(word)

Parameters:: word (str)
Return type:: int

vec(word)

Parameters:: word (str)
Return type:: Optional[numpy.ndarray]

count(word)

Parameters:: word (str)
Return type:: int

item(word)

Parameters:: word (str)
Return type:: WordDescriptor

__contains__(word)

Parameters:: word (str)
Return type:: bool

__eq__(other)

Parameters:: other (Any)
Return type:: bool

save(save_path, serialiser=AvailableSerialisers.dill, overwrite=False)

Save Vocab at path.

Parameters:

save_path (str) – The path to save at.
serialiser (Union[ str, AvailableSerialisers], optional) – The serialiser. Defaults to AvailableSerialisers.dill.
overwrite (bool, optional) – Whether to allow overwriting existing files. Defaults to False.

Return type:

None

classmethod load(path)

Parameters:: path (str)
Return type:: Vocab

get_strategy()

Return type:: SerialisingStrategy

classmethod get_init_attrs()

Return type:: list[str]

classmethod ignore_attrs()

Return type:: list[str]

classmethod include_properties()

Return type:: list[str]

class medcat.components.addons.relation_extraction.rel_cat.Config(/, **data)

Bases: SerialisableBaseModel

The base serialisable config.

Parameters:: data (Any)

general: General

components: Components

preprocessing: Preprocessing

cdb_maker: CDBMaker

annotation_output: AnnotationOutput

meta: ModelMeta

get_strategy()

Return type:: medcat.storage.serialisables.SerialisingStrategy

classmethod get_init_attrs()

Return type:: list[str]

classmethod ignore_attrs()

Return type:: list[str]

classmethod include_properties()

Return type:: list[str]

merge_config(other)

Merge this config with another config’s (partial) model dump.

The exepctation is that the other dict is a partial model dump. Values specified there are overwritten into the current config. Values not specified there are left intact.

The other config can have keys/values that do not exist in the config or sub-config. And they will be added where possible.

Parameters:: other (dict) – The model dump
Raises:: IncorrectConfigValues – If unable to set the attribute, trying to set incorrect value, or trying to set sub-config values in an incorrect format (non-dict).

classmethod load(path)

Parameters:: path (str)
Return type:: typing_extensions.Self

model_config: ClassVar[pydantic.config.ConfigDict]: Configuration for the model, should be a dictionary conforming to [ConfigDict][pydantic.config.ConfigDict].

model_fields: ClassVar[Dict[str, pydantic.fields.FieldInfo]]

Metadata about the fields defined on the model, mapping of field names to [FieldInfo][pydantic.fields.FieldInfo] objects.

This replaces Model.__fields__ from Pydantic V1.

model_computed_fields: ClassVar[Dict[str, pydantic.fields.ComputedFieldInfo]]: A dictionary of computed field names and their corresponding ComputedFieldInfo objects.

__class_vars__: ClassVar[set[str]]: The names of the class variables defined on the model.

__private_attributes__: ClassVar[Dict[str, pydantic.fields.ModelPrivateAttr]]: Metadata about the private attributes of the model.

__signature__: ClassVar[inspect.Signature]: The synthesized __init__ [Signature][inspect.Signature] of the model.

__pydantic_complete__: ClassVar[bool] = False: Whether model building is completed, or if there are still undefined fields.

__pydantic_core_schema__: ClassVar[pydantic_core.CoreSchema]: The core schema of the model.

__pydantic_custom_init__: ClassVar[bool]: Whether the model has a custom __init__ method.

__pydantic_decorators__: ClassVar[pydantic._internal._decorators.DecoratorInfos]: Metadata containing the decorators defined on the model. This replaces Model.__validators__ and Model.__root_validators__ from Pydantic V1.

__pydantic_generic_metadata__: ClassVar[pydantic._internal._generics.PydanticGenericMetadata]: Metadata for generic models; contains data used for a similar purpose to __args__, __origin__, __parameters__ in typing-module generics. May eventually be replaced by these.

__pydantic_parent_namespace__: ClassVar[Dict[str, Any] | None] = None: Parent namespace of the model, used for automatic rebuilding of models.

__pydantic_post_init__: ClassVar[None | Literal['model_post_init']]: The name of the post-init method for the model, if defined.

__pydantic_root_model__: ClassVar[bool] = False: Whether the model is a [RootModel][pydantic.root_model.RootModel].

__pydantic_serializer__: ClassVar[pydantic_core.SchemaSerializer]: The pydantic-core SchemaSerializer used to dump instances of the model.

__pydantic_validator__: ClassVar[pydantic_core.SchemaValidator | pydantic.plugin._schema_validator.PluggableSchemaValidator]: The pydantic-core SchemaValidator used to validate instances of the model.

__pydantic_extra__: dict[str, Any] | None: A dictionary containing extra values, if [extra][pydantic.config.ConfigDict.extra] is set to ‘allow’.

__pydantic_fields_set__: set[str]: The names of fields explicitly set during instantiation.

__pydantic_private__: dict[str, Any] | None: Values of private attributes set on the model instance.

__slots__ = ('__dict__', '__pydantic_fields_set__', '__pydantic_extra__', '__pydantic_private__')

__init__(/, **data)

Create a new model by parsing and validating input data from keyword arguments.

Raises [ValidationError][pydantic_core.ValidationError] if the input data cannot be validated to form a valid model.

self is explicitly positional-only to allow self as a field name.

Parameters:: data (Any)
Return type:: None

property model_extra: dict[str, Any] | None

Get extra fields set during validation.

Returns:: A dictionary of extra fields, or `None` if `config.extra` is not set to `”allow”`.
Return type:: dict[str, Any] | None

property model_fields_set: set[str]

Returns the set of fields that have been explicitly set on this model instance.

Returns:: A set of strings representing the fields that have been set, – i.e. that were not filled from defaults.
Return type:: set[str]

classmethod model_construct(_fields_set=None, **values)

Creates a new instance of the Model class with validated data.

Creates a new model setting __dict__ and __pydantic_fields_set__ from trusted or pre-validated data. Default values are respected, but no other validation is performed.

!!! note: model_construct() generally respects the model_config.extra setting on the provided model. That is, if model_config.extra == ‘allow’, then all extra passed values are added to the model instance’s __dict__ and __pydantic_extra__ fields. If model_config.extra == ‘ignore’ (the default), then all extra passed values are ignored. Because no validation is performed with a call to model_construct(), having model_config.extra == ‘forbid’ does not result in an error if extra values are passed, but they will be ignored.

Parameters:

_fields_set (set[str] | None) – A set of field names that were originally explicitly set during instantiation. If provided, this is directly used for the [model_fields_set][pydantic.BaseModel.model_fields_set] attribute. Otherwise, the field names from the values argument will be used.
values (Any) – Trusted or pre-validated data dictionary.

Returns:

A new instance of the `Model` class with validated data.

Return type:

typing_extensions.Self

model_copy(*, update=None, deep=False)

Usage docs: https://docs.pydantic.dev/2.9/concepts/serialization/#model_copy

Returns a copy of the model.

Parameters:

update (dict[str, Any] | None) – Values to change/add in the new model. Note: the data is not validated before creating the new model. You should trust this data.
deep (bool) – Set to True to make a deep copy of the model.

Returns:

New model instance.

Return type:

typing_extensions.Self

model_dump(*, mode='python', include=None, exclude=None, context=None, by_alias=False, exclude_unset=False, exclude_defaults=False, exclude_none=False, round_trip=False, warnings=True, serialize_as_any=False)

Usage docs: https://docs.pydantic.dev/2.9/concepts/serialization/#modelmodel_dump

Generate a dictionary representation of the model, optionally specifying which fields to include or exclude.

Parameters:

mode (Literal['json', 'python'] | str) – The mode in which to_python should run. If mode is ‘json’, the output will only contain JSON serializable types. If mode is ‘python’, the output may contain non-JSON-serializable Python objects.
include (IncEx | None) – A set of fields to include in the output.
exclude (IncEx | None) – A set of fields to exclude from the output.
context (Any | None) – Additional context to pass to the serializer.
by_alias (bool) – Whether to use the field’s alias in the dictionary key if defined.
exclude_unset (bool) – Whether to exclude fields that have not been explicitly set.
exclude_defaults (bool) – Whether to exclude fields that are set to their default value.
exclude_none (bool) – Whether to exclude fields that have a value of None.
round_trip (bool) – If True, dumped values should be valid as input for non-idempotent types such as Json[T].
warnings (bool | Literal['none', 'warn', 'error']) – How to handle serialization errors. False/”none” ignores them, True/”warn” logs errors, “error” raises a [PydanticSerializationError][pydantic_core.PydanticSerializationError].
serialize_as_any (bool) – Whether to serialize fields with duck-typing serialization behavior.

Returns:

A dictionary representation of the model.

Return type:

dict[str, Any]

model_dump_json(*, indent=None, include=None, exclude=None, context=None, by_alias=False, exclude_unset=False, exclude_defaults=False, exclude_none=False, round_trip=False, warnings=True, serialize_as_any=False)

Usage docs: https://docs.pydantic.dev/2.9/concepts/serialization/#modelmodel_dump_json

Generates a JSON representation of the model using Pydantic’s to_json method.

Parameters:

indent (int | None) – Indentation to use in the JSON output. If None is passed, the output will be compact.
include (IncEx | None) – Field(s) to include in the JSON output.
exclude (IncEx | None) – Field(s) to exclude from the JSON output.
context (Any | None) – Additional context to pass to the serializer.
by_alias (bool) – Whether to serialize using field aliases.
exclude_unset (bool) – Whether to exclude fields that have not been explicitly set.
exclude_defaults (bool) – Whether to exclude fields that are set to their default value.
exclude_none (bool) – Whether to exclude fields that have a value of None.
round_trip (bool) – If True, dumped values should be valid as input for non-idempotent types such as Json[T].
warnings (bool | Literal['none', 'warn', 'error']) – How to handle serialization errors. False/”none” ignores them, True/”warn” logs errors, “error” raises a [PydanticSerializationError][pydantic_core.PydanticSerializationError].
serialize_as_any (bool) – Whether to serialize fields with duck-typing serialization behavior.

Returns:

A JSON string representation of the model.

Return type:

str

classmethod model_json_schema(by_alias=True, ref_template=DEFAULT_REF_TEMPLATE, schema_generator=GenerateJsonSchema, mode='validation')

Generates a JSON schema for a model class.

Parameters:

by_alias (bool) – Whether to use attribute aliases or not.
ref_template (str) – The reference template.
schema_generator (type[pydantic.json_schema.GenerateJsonSchema]) – To override the logic used to generate the JSON schema, as a subclass of GenerateJsonSchema with your desired modifications
mode (pydantic.json_schema.JsonSchemaMode) – The mode in which to generate the schema.

Returns:

The JSON schema for the given model class.

Return type:

dict[str, Any]

classmethod model_parametrized_name(params)

Compute the class name for parametrizations of generic classes.

This method can be overridden to achieve a custom naming scheme for generic BaseModels.

Parameters:: params (tuple[type[Any], Ellipsis]) – Tuple of types of the class. Given a generic class Model with 2 type variables and a concrete model Model[str, int], the value (str, int) would be passed to params.
Returns:: String representing the new class where `params` are passed to `cls` as type variables.
Raises:: TypeError – Raised when trying to generate concrete names for non-generic models.
Return type:: str

model_post_init(__context)

Override this method to perform additional initialization after __init__ and model_construct. This is useful if you want to do some validation that requires the entire model to be initialized.

Parameters:: __context (Any)
Return type:: None

classmethod model_rebuild(*, force=False, raise_errors=True, _parent_namespace_depth=2, _types_namespace=None)

Try to rebuild the pydantic-core schema for the model.

This may be necessary when one of the annotations is a ForwardRef which could not be resolved during the initial attempt to build the schema, and automatic rebuilding fails.

Parameters:

force (bool) – Whether to force the rebuilding of the model schema, defaults to False.
raise_errors (bool) – Whether to raise errors, defaults to True.
_parent_namespace_depth (int) – The depth level of the parent namespace, defaults to 2.
_types_namespace (dict[str, Any] | None) – The types namespace, defaults to None.

Returns:

Returns `None` if the schema is already “complete” and rebuilding was not required.
If rebuilding _was_ required, returns `True` if rebuilding was successful, otherwise `False`.

Return type:

bool | None

classmethod model_validate(obj, *, strict=None, from_attributes=None, context=None)

Validate a pydantic model instance.

Parameters:

obj (Any) – The object to validate.
strict (bool | None) – Whether to enforce types strictly.
from_attributes (bool | None) – Whether to extract data from object attributes.
context (Any | None) – Additional context to pass to the validator.

Raises:

ValidationError – If the object could not be validated.

Returns:

The validated model instance.

Return type:

typing_extensions.Self

classmethod model_validate_json(json_data, *, strict=None, context=None)

Usage docs: https://docs.pydantic.dev/2.9/concepts/json/#json-parsing

Validate the given JSON data against the Pydantic model.

Parameters:

json_data (str | bytes | bytearray) – The JSON data to validate.
strict (bool | None) – Whether to enforce types strictly.
context (Any | None) – Extra variables to pass to the validator.

Returns:

The validated Pydantic model.

Raises:

ValidationError – If json_data is not a JSON string or the object could not be validated.

Return type:

typing_extensions.Self

classmethod model_validate_strings(obj, *, strict=None, context=None)

Validate the given object with string data against the Pydantic model.

Parameters:

obj (Any) – The object containing string data to validate.
strict (bool | None) – Whether to enforce types strictly.
context (Any | None) – Extra variables to pass to the validator.

Returns:

The validated Pydantic model.

Return type:

typing_extensions.Self

classmethod __get_pydantic_core_schema__(source, handler, /)

Hook into generating the model’s CoreSchema.

Parameters:

source (type[BaseModel]) – The class we are generating a schema for. This will generally be the same as the cls argument if this is a classmethod.
handler (pydantic.annotated_handlers.GetCoreSchemaHandler) – A callable that calls into Pydantic’s internal CoreSchema generation logic.

Returns:

A `pydantic-core` `CoreSchema`.

Return type:

pydantic_core.CoreSchema

classmethod __get_pydantic_json_schema__(core_schema, handler, /)

Hook into generating the model’s JSON schema.

Parameters:

core_schema (pydantic_core.CoreSchema) – A pydantic-core CoreSchema. You can ignore this argument and call the handler with a new CoreSchema, wrap this CoreSchema ({‘type’: ‘nullable’, ‘schema’: current_schema}), or just call the handler with the original schema.
handler (pydantic.annotated_handlers.GetJsonSchemaHandler) – Call into Pydantic’s internal JSON schema generation. This will raise a pydantic.errors.PydanticInvalidForJsonSchema if JSON schema generation fails. Since this gets called by BaseModel.model_json_schema you can override the schema_generator argument to that function to change JSON schema generation globally for a type.

Returns:

A JSON schema, as a Python object.

Return type:

pydantic.json_schema.JsonSchemaValue

classmethod __pydantic_init_subclass__(**kwargs)

This is intended to behave just like __init_subclass__, but is called by ModelMetaclass only after the class is actually fully initialized. In particular, attributes like model_fields will be present when this is called.

This is necessary because __init_subclass__ will always be called by type.__new__, and it would require a prohibitively large refactor to the ModelMetaclass to ensure that type.__new__ was called in such a manner that the class would already be sufficiently initialized.

This will receive the same kwargs that would be passed to the standard __init_subclass__, namely, any kwargs passed to the class definition that aren’t used internally by pydantic.

Parameters:: **kwargs (Any) – Any keyword arguments passed to the class definition that aren’t used internally by pydantic.
Return type:: None

classmethod __class_getitem__(typevar_values)

Parameters:: typevar_values (type[Any] | tuple[type[Any], Ellipsis])
Return type:: type[BaseModel] | pydantic._internal._forward_ref.PydanticRecursiveRef

__copy__()

Returns a shallow copy of the model.

Return type:: typing_extensions.Self

__deepcopy__(memo=None)

Returns a deep copy of the model.

Parameters:: memo (dict[int, Any] | None)
Return type:: typing_extensions.Self

__getattr__(item)

Parameters:: item (str)
Return type:: Any

_check_frozen(name, value)

Parameters:

name (str)
value (Any)

Return type:

None

__getstate__()

Return type:: dict[Any, Any]

__setstate__(state)

Parameters:: state (dict[Any, Any])
Return type:: None

__eq__(other)

Parameters:: other (Any)
Return type:: bool

classmethod __init_subclass__(**kwargs)

This signature is included purely to help type-checkers check arguments to class declaration, which provides a way to conveniently set model_config key/value pairs.

```py from pydantic import BaseModel

class MyModel(BaseModel, extra=’allow’): … ```

However, this may be deceiving, since the _actual_ calls to __init_subclass__ will not receive any of the config arguments, and will only receive any keyword arguments passed during class initialization that are _not_ expected keys in ConfigDict. (This is due to the way ModelMetaclass.__new__ works.)

Parameters:: **kwargs (typing_extensions.Unpack[pydantic.config.ConfigDict]) – Keyword arguments passed to the class definition, which set model_config

Note

You may want to override __pydantic_init_subclass__ instead, which behaves similarly but is called after the class is fully initialized.

__iter__()

So dict(model) works.

Return type:: TupleGenerator

__repr__()

Return type:: str

__repr_args__()

Return type:: pydantic._internal._repr.ReprArgs

__repr_name__

__repr_str__

__pretty__

__rich_repr__

__str__()

Return type:: str

property __fields__: dict[str, pydantic.fields.FieldInfo]

Return type:: dict[str, pydantic.fields.FieldInfo]

property __fields_set__: set[str]

Return type:: set[str]

dict(*, include=None, exclude=None, by_alias=False, exclude_unset=False, exclude_defaults=False, exclude_none=False)

Parameters:

include (IncEx | None)
exclude (IncEx | None)
by_alias (bool)
exclude_unset (bool)
exclude_defaults (bool)
exclude_none (bool)

Return type:

Dict[str, Any]

json(*, include=None, exclude=None, by_alias=False, exclude_unset=False, exclude_defaults=False, exclude_none=False, encoder=PydanticUndefined, models_as_dict=PydanticUndefined, **dumps_kwargs)

Parameters:

include (IncEx | None)
exclude (IncEx | None)
by_alias (bool)
exclude_unset (bool)
exclude_defaults (bool)
exclude_none (bool)
encoder (Callable[[Any], Any] | None)
models_as_dict (bool)
dumps_kwargs (Any)

Return type:

str

classmethod parse_obj(obj)

Parameters:: obj (Any)
Return type:: typing_extensions.Self

classmethod parse_raw(b, *, content_type=None, encoding='utf8', proto=None, allow_pickle=False)

Parameters:

b (str | bytes)
content_type (str | None)
encoding (str)
proto (pydantic.deprecated.parse.Protocol | None)
allow_pickle (bool)

Return type:

typing_extensions.Self

classmethod parse_file(path, *, content_type=None, encoding='utf8', proto=None, allow_pickle=False)

Parameters:

path (str | pathlib.Path)
content_type (str | None)
encoding (str)
proto (pydantic.deprecated.parse.Protocol | None)
allow_pickle (bool)

Return type:

typing_extensions.Self

classmethod from_orm(obj)

Parameters:: obj (Any)
Return type:: typing_extensions.Self

classmethod construct(_fields_set=None, **values)

Parameters:

_fields_set (set[str] | None)
values (Any)

Return type:

typing_extensions.Self

copy(*, include=None, exclude=None, update=None, deep=False)

Returns a copy of the model.

!!! warning “Deprecated”: This method is now deprecated; use model_copy instead.

If you need include or exclude, use:

`py data = self.model_dump(include=include, exclude=exclude, round_trip=True) data = {**data, **(update or {})} copied = self.model_validate(data) `

Parameters:

include (pydantic._internal._utils.AbstractSetIntStr | pydantic._internal._utils.MappingIntStrAny | None) – Optional set or mapping specifying which fields to include in the copied model.
exclude (pydantic._internal._utils.AbstractSetIntStr | pydantic._internal._utils.MappingIntStrAny | None) – Optional set or mapping specifying which fields to exclude in the copied model.
update (Dict[str, Any] | None) – Optional dictionary of field-value pairs to override field values in the copied model.
deep (bool) – If True, the values of fields that are Pydantic models will be deep-copied.

Returns:

A copy of the model with included, excluded and updated fields as specified.

Return type:

typing_extensions.Self

classmethod schema(by_alias=True, ref_template=DEFAULT_REF_TEMPLATE)

Parameters:

by_alias (bool)
ref_template (str)

Return type:

Dict[str, Any]

classmethod schema_json(*, by_alias=True, ref_template=DEFAULT_REF_TEMPLATE, **dumps_kwargs)

Parameters:

by_alias (bool)
ref_template (str)
dumps_kwargs (Any)

Return type:

str

classmethod validate(value)

Parameters:: value (Any)
Return type:: typing_extensions.Self

classmethod update_forward_refs(**localns)

Parameters:: localns (Any)
Return type:: None

_iter(*args, **kwargs)

Parameters:

args (Any)
kwargs (Any)

Return type:

Any

_copy_and_set_values(*args, **kwargs)

Parameters:

args (Any)
kwargs (Any)

Return type:

Any

classmethod _get_value(*args, **kwargs)

Parameters:

args (Any)
kwargs (Any)

Return type:

Any

_calculate_keys(*args, **kwargs)

Parameters:

args (Any)
kwargs (Any)

Return type:

Any

class medcat.components.addons.relation_extraction.rel_cat.ComponentConfig(/, **data)

Bases: DirtiableBaseModel

The base serialisable config.

Parameters:: data (Any)

comp_name: str = 'default'

The name of the component.

If a custom implementation is required, it needs to be registered using `medcat.components.types.register_core_component(

<core component type>, <component name>, <implementing class>)

By default, only the ‘default’ component is registered.

_is_dirty: bool = False

__setattr__(name, value)

Parameters:

name (str)
value (Any)

property is_dirty: bool

Return type:: bool

mark_clean()

get_strategy()

Return type:: medcat.storage.serialisables.SerialisingStrategy

classmethod get_init_attrs()

Return type:: list[str]

classmethod ignore_attrs()

Return type:: list[str]

classmethod include_properties()

Return type:: list[str]

merge_config(other)

Merge this config with another config’s (partial) model dump.

The exepctation is that the other dict is a partial model dump. Values specified there are overwritten into the current config. Values not specified there are left intact.

The other config can have keys/values that do not exist in the config or sub-config. And they will be added where possible.

Parameters:: other (dict) – The model dump
Raises:: IncorrectConfigValues – If unable to set the attribute, trying to set incorrect value, or trying to set sub-config values in an incorrect format (non-dict).

classmethod load(path)

Parameters:: path (str)
Return type:: typing_extensions.Self

model_config: ClassVar[pydantic.config.ConfigDict]: Configuration for the model, should be a dictionary conforming to [ConfigDict][pydantic.config.ConfigDict].

model_fields: ClassVar[Dict[str, pydantic.fields.FieldInfo]]

Metadata about the fields defined on the model, mapping of field names to [FieldInfo][pydantic.fields.FieldInfo] objects.

This replaces Model.__fields__ from Pydantic V1.

model_computed_fields: ClassVar[Dict[str, pydantic.fields.ComputedFieldInfo]]: A dictionary of computed field names and their corresponding ComputedFieldInfo objects.

__class_vars__: ClassVar[set[str]]: The names of the class variables defined on the model.

__private_attributes__: ClassVar[Dict[str, pydantic.fields.ModelPrivateAttr]]: Metadata about the private attributes of the model.

__signature__: ClassVar[inspect.Signature]: The synthesized __init__ [Signature][inspect.Signature] of the model.

__pydantic_complete__: ClassVar[bool] = False: Whether model building is completed, or if there are still undefined fields.

__pydantic_core_schema__: ClassVar[pydantic_core.CoreSchema]: The core schema of the model.

__pydantic_custom_init__: ClassVar[bool]: Whether the model has a custom __init__ method.

__pydantic_decorators__: ClassVar[pydantic._internal._decorators.DecoratorInfos]: Metadata containing the decorators defined on the model. This replaces Model.__validators__ and Model.__root_validators__ from Pydantic V1.

__pydantic_generic_metadata__: ClassVar[pydantic._internal._generics.PydanticGenericMetadata]: Metadata for generic models; contains data used for a similar purpose to __args__, __origin__, __parameters__ in typing-module generics. May eventually be replaced by these.

__pydantic_parent_namespace__: ClassVar[Dict[str, Any] | None] = None: Parent namespace of the model, used for automatic rebuilding of models.

__pydantic_post_init__: ClassVar[None | Literal['model_post_init']]: The name of the post-init method for the model, if defined.

__pydantic_root_model__: ClassVar[bool] = False: Whether the model is a [RootModel][pydantic.root_model.RootModel].

__pydantic_serializer__: ClassVar[pydantic_core.SchemaSerializer]: The pydantic-core SchemaSerializer used to dump instances of the model.

__pydantic_validator__: ClassVar[pydantic_core.SchemaValidator | pydantic.plugin._schema_validator.PluggableSchemaValidator]: The pydantic-core SchemaValidator used to validate instances of the model.

__pydantic_extra__: dict[str, Any] | None: A dictionary containing extra values, if [extra][pydantic.config.ConfigDict.extra] is set to ‘allow’.

__pydantic_fields_set__: set[str]: The names of fields explicitly set during instantiation.

__pydantic_private__: dict[str, Any] | None: Values of private attributes set on the model instance.

__slots__ = ('__dict__', '__pydantic_fields_set__', '__pydantic_extra__', '__pydantic_private__')

__init__(/, **data)

Create a new model by parsing and validating input data from keyword arguments.

Raises [ValidationError][pydantic_core.ValidationError] if the input data cannot be validated to form a valid model.

self is explicitly positional-only to allow self as a field name.

Parameters:: data (Any)
Return type:: None

property model_extra: dict[str, Any] | None

Get extra fields set during validation.

Returns:: A dictionary of extra fields, or `None` if `config.extra` is not set to `”allow”`.
Return type:: dict[str, Any] | None

property model_fields_set: set[str]

Returns the set of fields that have been explicitly set on this model instance.

Returns:: A set of strings representing the fields that have been set, – i.e. that were not filled from defaults.
Return type:: set[str]

classmethod model_construct(_fields_set=None, **values)

Creates a new instance of the Model class with validated data.

Creates a new model setting __dict__ and __pydantic_fields_set__ from trusted or pre-validated data. Default values are respected, but no other validation is performed.

!!! note: model_construct() generally respects the model_config.extra setting on the provided model. That is, if model_config.extra == ‘allow’, then all extra passed values are added to the model instance’s __dict__ and __pydantic_extra__ fields. If model_config.extra == ‘ignore’ (the default), then all extra passed values are ignored. Because no validation is performed with a call to model_construct(), having model_config.extra == ‘forbid’ does not result in an error if extra values are passed, but they will be ignored.

Parameters:

_fields_set (set[str] | None) – A set of field names that were originally explicitly set during instantiation. If provided, this is directly used for the [model_fields_set][pydantic.BaseModel.model_fields_set] attribute. Otherwise, the field names from the values argument will be used.
values (Any) – Trusted or pre-validated data dictionary.

Returns:

A new instance of the `Model` class with validated data.

Return type:

typing_extensions.Self

model_copy(*, update=None, deep=False)

Usage docs: https://docs.pydantic.dev/2.9/concepts/serialization/#model_copy

Returns a copy of the model.

Parameters:

update (dict[str, Any] | None) – Values to change/add in the new model. Note: the data is not validated before creating the new model. You should trust this data.
deep (bool) – Set to True to make a deep copy of the model.

Returns:

New model instance.

Return type:

typing_extensions.Self

model_dump(*, mode='python', include=None, exclude=None, context=None, by_alias=False, exclude_unset=False, exclude_defaults=False, exclude_none=False, round_trip=False, warnings=True, serialize_as_any=False)

Usage docs: https://docs.pydantic.dev/2.9/concepts/serialization/#modelmodel_dump

Generate a dictionary representation of the model, optionally specifying which fields to include or exclude.

Parameters:

mode (Literal['json', 'python'] | str) – The mode in which to_python should run. If mode is ‘json’, the output will only contain JSON serializable types. If mode is ‘python’, the output may contain non-JSON-serializable Python objects.
include (IncEx | None) – A set of fields to include in the output.
exclude (IncEx | None) – A set of fields to exclude from the output.
context (Any | None) – Additional context to pass to the serializer.
by_alias (bool) – Whether to use the field’s alias in the dictionary key if defined.
exclude_unset (bool) – Whether to exclude fields that have not been explicitly set.
exclude_defaults (bool) – Whether to exclude fields that are set to their default value.
exclude_none (bool) – Whether to exclude fields that have a value of None.
round_trip (bool) – If True, dumped values should be valid as input for non-idempotent types such as Json[T].
warnings (bool | Literal['none', 'warn', 'error']) – How to handle serialization errors. False/”none” ignores them, True/”warn” logs errors, “error” raises a [PydanticSerializationError][pydantic_core.PydanticSerializationError].
serialize_as_any (bool) – Whether to serialize fields with duck-typing serialization behavior.

Returns:

A dictionary representation of the model.

Return type:

dict[str, Any]

model_dump_json(*, indent=None, include=None, exclude=None, context=None, by_alias=False, exclude_unset=False, exclude_defaults=False, exclude_none=False, round_trip=False, warnings=True, serialize_as_any=False)

Usage docs: https://docs.pydantic.dev/2.9/concepts/serialization/#modelmodel_dump_json

Generates a JSON representation of the model using Pydantic’s to_json method.

Parameters:

indent (int | None) – Indentation to use in the JSON output. If None is passed, the output will be compact.
include (IncEx | None) – Field(s) to include in the JSON output.
exclude (IncEx | None) – Field(s) to exclude from the JSON output.
context (Any | None) – Additional context to pass to the serializer.
by_alias (bool) – Whether to serialize using field aliases.
exclude_unset (bool) – Whether to exclude fields that have not been explicitly set.
exclude_defaults (bool) – Whether to exclude fields that are set to their default value.
exclude_none (bool) – Whether to exclude fields that have a value of None.
round_trip (bool) – If True, dumped values should be valid as input for non-idempotent types such as Json[T].
warnings (bool | Literal['none', 'warn', 'error']) – How to handle serialization errors. False/”none” ignores them, True/”warn” logs errors, “error” raises a [PydanticSerializationError][pydantic_core.PydanticSerializationError].
serialize_as_any (bool) – Whether to serialize fields with duck-typing serialization behavior.

Returns:

A JSON string representation of the model.

Return type:

str

classmethod model_json_schema(by_alias=True, ref_template=DEFAULT_REF_TEMPLATE, schema_generator=GenerateJsonSchema, mode='validation')

Generates a JSON schema for a model class.

Parameters:

by_alias (bool) – Whether to use attribute aliases or not.
ref_template (str) – The reference template.
schema_generator (type[pydantic.json_schema.GenerateJsonSchema]) – To override the logic used to generate the JSON schema, as a subclass of GenerateJsonSchema with your desired modifications
mode (pydantic.json_schema.JsonSchemaMode) – The mode in which to generate the schema.

Returns:

The JSON schema for the given model class.

Return type:

dict[str, Any]

classmethod model_parametrized_name(params)

Compute the class name for parametrizations of generic classes.

This method can be overridden to achieve a custom naming scheme for generic BaseModels.

Parameters:: params (tuple[type[Any], Ellipsis]) – Tuple of types of the class. Given a generic class Model with 2 type variables and a concrete model Model[str, int], the value (str, int) would be passed to params.
Returns:: String representing the new class where `params` are passed to `cls` as type variables.
Raises:: TypeError – Raised when trying to generate concrete names for non-generic models.
Return type:: str

model_post_init(__context)

Override this method to perform additional initialization after __init__ and model_construct. This is useful if you want to do some validation that requires the entire model to be initialized.

Parameters:: __context (Any)
Return type:: None

classmethod model_rebuild(*, force=False, raise_errors=True, _parent_namespace_depth=2, _types_namespace=None)

Try to rebuild the pydantic-core schema for the model.

This may be necessary when one of the annotations is a ForwardRef which could not be resolved during the initial attempt to build the schema, and automatic rebuilding fails.

Parameters:

force (bool) – Whether to force the rebuilding of the model schema, defaults to False.
raise_errors (bool) – Whether to raise errors, defaults to True.
_parent_namespace_depth (int) – The depth level of the parent namespace, defaults to 2.
_types_namespace (dict[str, Any] | None) – The types namespace, defaults to None.

Returns:

Returns `None` if the schema is already “complete” and rebuilding was not required.
If rebuilding _was_ required, returns `True` if rebuilding was successful, otherwise `False`.

Return type:

bool | None

classmethod model_validate(obj, *, strict=None, from_attributes=None, context=None)

Validate a pydantic model instance.

Parameters:

obj (Any) – The object to validate.
strict (bool | None) – Whether to enforce types strictly.
from_attributes (bool | None) – Whether to extract data from object attributes.
context (Any | None) – Additional context to pass to the validator.

Raises:

ValidationError – If the object could not be validated.

Returns:

The validated model instance.

Return type:

typing_extensions.Self

classmethod model_validate_json(json_data, *, strict=None, context=None)

Usage docs: https://docs.pydantic.dev/2.9/concepts/json/#json-parsing

Validate the given JSON data against the Pydantic model.

Parameters:

json_data (str | bytes | bytearray) – The JSON data to validate.
strict (bool | None) – Whether to enforce types strictly.
context (Any | None) – Extra variables to pass to the validator.

Returns:

The validated Pydantic model.

Raises:

ValidationError – If json_data is not a JSON string or the object could not be validated.

Return type:

typing_extensions.Self

classmethod model_validate_strings(obj, *, strict=None, context=None)

Validate the given object with string data against the Pydantic model.

Parameters:

obj (Any) – The object containing string data to validate.
strict (bool | None) – Whether to enforce types strictly.
context (Any | None) – Extra variables to pass to the validator.

Returns:

The validated Pydantic model.

Return type:

typing_extensions.Self

classmethod __get_pydantic_core_schema__(source, handler, /)

Hook into generating the model’s CoreSchema.

Parameters:

source (type[BaseModel]) – The class we are generating a schema for. This will generally be the same as the cls argument if this is a classmethod.
handler (pydantic.annotated_handlers.GetCoreSchemaHandler) – A callable that calls into Pydantic’s internal CoreSchema generation logic.

Returns:

A `pydantic-core` `CoreSchema`.

Return type:

pydantic_core.CoreSchema

classmethod __get_pydantic_json_schema__(core_schema, handler, /)

Hook into generating the model’s JSON schema.

Parameters:

core_schema (pydantic_core.CoreSchema) – A pydantic-core CoreSchema. You can ignore this argument and call the handler with a new CoreSchema, wrap this CoreSchema ({‘type’: ‘nullable’, ‘schema’: current_schema}), or just call the handler with the original schema.
handler (pydantic.annotated_handlers.GetJsonSchemaHandler) – Call into Pydantic’s internal JSON schema generation. This will raise a pydantic.errors.PydanticInvalidForJsonSchema if JSON schema generation fails. Since this gets called by BaseModel.model_json_schema you can override the schema_generator argument to that function to change JSON schema generation globally for a type.

Returns:

A JSON schema, as a Python object.

Return type:

pydantic.json_schema.JsonSchemaValue

classmethod __pydantic_init_subclass__(**kwargs)

This is intended to behave just like __init_subclass__, but is called by ModelMetaclass only after the class is actually fully initialized. In particular, attributes like model_fields will be present when this is called.

This is necessary because __init_subclass__ will always be called by type.__new__, and it would require a prohibitively large refactor to the ModelMetaclass to ensure that type.__new__ was called in such a manner that the class would already be sufficiently initialized.

This will receive the same kwargs that would be passed to the standard __init_subclass__, namely, any kwargs passed to the class definition that aren’t used internally by pydantic.

Parameters:: **kwargs (Any) – Any keyword arguments passed to the class definition that aren’t used internally by pydantic.
Return type:: None

classmethod __class_getitem__(typevar_values)

Parameters:: typevar_values (type[Any] | tuple[type[Any], Ellipsis])
Return type:: type[BaseModel] | pydantic._internal._forward_ref.PydanticRecursiveRef

__copy__()

Returns a shallow copy of the model.

Return type:: typing_extensions.Self

__deepcopy__(memo=None)

Returns a deep copy of the model.

Parameters:: memo (dict[int, Any] | None)
Return type:: typing_extensions.Self

__getattr__(item)

Parameters:: item (str)
Return type:: Any

_check_frozen(name, value)

Parameters:

name (str)
value (Any)

Return type:

None

__getstate__()

Return type:: dict[Any, Any]

__setstate__(state)

Parameters:: state (dict[Any, Any])
Return type:: None

__eq__(other)

Parameters:: other (Any)
Return type:: bool

classmethod __init_subclass__(**kwargs)

This signature is included purely to help type-checkers check arguments to class declaration, which provides a way to conveniently set model_config key/value pairs.

```py from pydantic import BaseModel

class MyModel(BaseModel, extra=’allow’): … ```

However, this may be deceiving, since the _actual_ calls to __init_subclass__ will not receive any of the config arguments, and will only receive any keyword arguments passed during class initialization that are _not_ expected keys in ConfigDict. (This is due to the way ModelMetaclass.__new__ works.)

Parameters:: **kwargs (typing_extensions.Unpack[pydantic.config.ConfigDict]) – Keyword arguments passed to the class definition, which set model_config

Note

You may want to override __pydantic_init_subclass__ instead, which behaves similarly but is called after the class is fully initialized.

__iter__()

So dict(model) works.

Return type:: TupleGenerator

__repr__()

Return type:: str

__repr_args__()

Return type:: pydantic._internal._repr.ReprArgs

__repr_name__

__repr_str__

__pretty__

__rich_repr__

__str__()

Return type:: str

property __fields__: dict[str, pydantic.fields.FieldInfo]

Return type:: dict[str, pydantic.fields.FieldInfo]

property __fields_set__: set[str]

Return type:: set[str]

dict(*, include=None, exclude=None, by_alias=False, exclude_unset=False, exclude_defaults=False, exclude_none=False)

Parameters:

include (IncEx | None)
exclude (IncEx | None)
by_alias (bool)
exclude_unset (bool)
exclude_defaults (bool)
exclude_none (bool)

Return type:

Dict[str, Any]

json(*, include=None, exclude=None, by_alias=False, exclude_unset=False, exclude_defaults=False, exclude_none=False, encoder=PydanticUndefined, models_as_dict=PydanticUndefined, **dumps_kwargs)

Parameters:

include (IncEx | None)
exclude (IncEx | None)
by_alias (bool)
exclude_unset (bool)
exclude_defaults (bool)
exclude_none (bool)
encoder (Callable[[Any], Any] | None)
models_as_dict (bool)
dumps_kwargs (Any)

Return type:

str

classmethod parse_obj(obj)

Parameters:: obj (Any)
Return type:: typing_extensions.Self

classmethod parse_raw(b, *, content_type=None, encoding='utf8', proto=None, allow_pickle=False)

Parameters:

b (str | bytes)
content_type (str | None)
encoding (str)
proto (pydantic.deprecated.parse.Protocol | None)
allow_pickle (bool)

Return type:

typing_extensions.Self

classmethod parse_file(path, *, content_type=None, encoding='utf8', proto=None, allow_pickle=False)

Parameters:

path (str | pathlib.Path)
content_type (str | None)
encoding (str)
proto (pydantic.deprecated.parse.Protocol | None)
allow_pickle (bool)

Return type:

typing_extensions.Self

classmethod from_orm(obj)

Parameters:: obj (Any)
Return type:: typing_extensions.Self

classmethod construct(_fields_set=None, **values)

Parameters:

_fields_set (set[str] | None)
values (Any)

Return type:

typing_extensions.Self

copy(*, include=None, exclude=None, update=None, deep=False)

Returns a copy of the model.

!!! warning “Deprecated”: This method is now deprecated; use model_copy instead.

If you need include or exclude, use:

`py data = self.model_dump(include=include, exclude=exclude, round_trip=True) data = {**data, **(update or {})} copied = self.model_validate(data) `

Parameters:

include (pydantic._internal._utils.AbstractSetIntStr | pydantic._internal._utils.MappingIntStrAny | None) – Optional set or mapping specifying which fields to include in the copied model.
exclude (pydantic._internal._utils.AbstractSetIntStr | pydantic._internal._utils.MappingIntStrAny | None) – Optional set or mapping specifying which fields to exclude in the copied model.
update (Dict[str, Any] | None) – Optional dictionary of field-value pairs to override field values in the copied model.
deep (bool) – If True, the values of fields that are Pydantic models will be deep-copied.

Returns:

A copy of the model with included, excluded and updated fields as specified.

Return type:

typing_extensions.Self

classmethod schema(by_alias=True, ref_template=DEFAULT_REF_TEMPLATE)

Parameters:

by_alias (bool)
ref_template (str)

Return type:

Dict[str, Any]

classmethod schema_json(*, by_alias=True, ref_template=DEFAULT_REF_TEMPLATE, **dumps_kwargs)

Parameters:

by_alias (bool)
ref_template (str)
dumps_kwargs (Any)

Return type:

str

classmethod validate(value)

Parameters:: value (Any)
Return type:: typing_extensions.Self

classmethod update_forward_refs(**localns)

Parameters:: localns (Any)
Return type:: None

_iter(*args, **kwargs)

Parameters:

args (Any)
kwargs (Any)

Return type:

Any

_copy_and_set_values(*args, **kwargs)

Parameters:

args (Any)
kwargs (Any)

Return type:

Any

classmethod _get_value(*args, **kwargs)

Parameters:

args (Any)
kwargs (Any)

Return type:

Any

_calculate_keys(*args, **kwargs)

Parameters:

args (Any)
kwargs (Any)

Return type:

Any

class medcat.components.addons.relation_extraction.rel_cat.ConfigRelCAT(/, **data)

Bases: medcat.config.config.ComponentConfig

The RelCAT part of the config

Parameters:: data (Any)

general: General

model: Model

train: Train

class Config

extra = 'allow'

validate_assignment = True

classmethod load(load_path='./')

Load the config from a file.

Parameters:: load_path (str) – Path to RelCAT config. Defaults to “./”.
Returns:: ConfigRelCAT – The loaded config.
Return type:: ConfigRelCAT

comp_name: str = 'default'

The name of the component.

If a custom implementation is required, it needs to be registered using `medcat.components.types.register_core_component(

<core component type>, <component name>, <implementing class>)

By default, only the ‘default’ component is registered.

_is_dirty: bool = False

__setattr__(name, value)

Parameters:

name (str)
value (Any)

property is_dirty: bool

Return type:: bool

mark_clean()

get_strategy()

Return type:: medcat.storage.serialisables.SerialisingStrategy

classmethod get_init_attrs()

Return type:: list[str]

classmethod ignore_attrs()

Return type:: list[str]

classmethod include_properties()

Return type:: list[str]

merge_config(other)

Merge this config with another config’s (partial) model dump.

The exepctation is that the other dict is a partial model dump. Values specified there are overwritten into the current config. Values not specified there are left intact.

The other config can have keys/values that do not exist in the config or sub-config. And they will be added where possible.

Parameters:: other (dict) – The model dump
Raises:: IncorrectConfigValues – If unable to set the attribute, trying to set incorrect value, or trying to set sub-config values in an incorrect format (non-dict).

model_config: ClassVar[pydantic.config.ConfigDict]: Configuration for the model, should be a dictionary conforming to [ConfigDict][pydantic.config.ConfigDict].

model_fields: ClassVar[Dict[str, pydantic.fields.FieldInfo]]

Metadata about the fields defined on the model, mapping of field names to [FieldInfo][pydantic.fields.FieldInfo] objects.

This replaces Model.__fields__ from Pydantic V1.

model_computed_fields: ClassVar[Dict[str, pydantic.fields.ComputedFieldInfo]]: A dictionary of computed field names and their corresponding ComputedFieldInfo objects.

__class_vars__: ClassVar[set[str]]: The names of the class variables defined on the model.

__private_attributes__: ClassVar[Dict[str, pydantic.fields.ModelPrivateAttr]]: Metadata about the private attributes of the model.

__signature__: ClassVar[inspect.Signature]: The synthesized __init__ [Signature][inspect.Signature] of the model.

__pydantic_complete__: ClassVar[bool] = False: Whether model building is completed, or if there are still undefined fields.

__pydantic_core_schema__: ClassVar[pydantic_core.CoreSchema]: The core schema of the model.

__pydantic_custom_init__: ClassVar[bool]: Whether the model has a custom __init__ method.

__pydantic_decorators__: ClassVar[pydantic._internal._decorators.DecoratorInfos]: Metadata containing the decorators defined on the model. This replaces Model.__validators__ and Model.__root_validators__ from Pydantic V1.

__pydantic_generic_metadata__: ClassVar[pydantic._internal._generics.PydanticGenericMetadata]: Metadata for generic models; contains data used for a similar purpose to __args__, __origin__, __parameters__ in typing-module generics. May eventually be replaced by these.

__pydantic_parent_namespace__: ClassVar[Dict[str, Any] | None] = None: Parent namespace of the model, used for automatic rebuilding of models.

__pydantic_post_init__: ClassVar[None | Literal['model_post_init']]: The name of the post-init method for the model, if defined.

__pydantic_root_model__: ClassVar[bool] = False: Whether the model is a [RootModel][pydantic.root_model.RootModel].

__pydantic_serializer__: ClassVar[pydantic_core.SchemaSerializer]: The pydantic-core SchemaSerializer used to dump instances of the model.

__pydantic_validator__: ClassVar[pydantic_core.SchemaValidator | pydantic.plugin._schema_validator.PluggableSchemaValidator]: The pydantic-core SchemaValidator used to validate instances of the model.

__pydantic_extra__: dict[str, Any] | None: A dictionary containing extra values, if [extra][pydantic.config.ConfigDict.extra] is set to ‘allow’.

__pydantic_fields_set__: set[str]: The names of fields explicitly set during instantiation.

__pydantic_private__: dict[str, Any] | None: Values of private attributes set on the model instance.

__slots__ = ('__dict__', '__pydantic_fields_set__', '__pydantic_extra__', '__pydantic_private__')

__init__(/, **data)

Create a new model by parsing and validating input data from keyword arguments.

Raises [ValidationError][pydantic_core.ValidationError] if the input data cannot be validated to form a valid model.

self is explicitly positional-only to allow self as a field name.

Parameters:: data (Any)
Return type:: None

property model_extra: dict[str, Any] | None

Get extra fields set during validation.

Returns:: A dictionary of extra fields, or `None` if `config.extra` is not set to `”allow”`.
Return type:: dict[str, Any] | None

property model_fields_set: set[str]

Returns the set of fields that have been explicitly set on this model instance.

Returns:: A set of strings representing the fields that have been set, – i.e. that were not filled from defaults.
Return type:: set[str]

classmethod model_construct(_fields_set=None, **values)

Creates a new instance of the Model class with validated data.

Creates a new model setting __dict__ and __pydantic_fields_set__ from trusted or pre-validated data. Default values are respected, but no other validation is performed.

!!! note: model_construct() generally respects the model_config.extra setting on the provided model. That is, if model_config.extra == ‘allow’, then all extra passed values are added to the model instance’s __dict__ and __pydantic_extra__ fields. If model_config.extra == ‘ignore’ (the default), then all extra passed values are ignored. Because no validation is performed with a call to model_construct(), having model_config.extra == ‘forbid’ does not result in an error if extra values are passed, but they will be ignored.

Parameters:

_fields_set (set[str] | None) – A set of field names that were originally explicitly set during instantiation. If provided, this is directly used for the [model_fields_set][pydantic.BaseModel.model_fields_set] attribute. Otherwise, the field names from the values argument will be used.
values (Any) – Trusted or pre-validated data dictionary.

Returns:

A new instance of the `Model` class with validated data.

Return type:

typing_extensions.Self

model_copy(*, update=None, deep=False)

Usage docs: https://docs.pydantic.dev/2.9/concepts/serialization/#model_copy

Returns a copy of the model.

Parameters:

update (dict[str, Any] | None) – Values to change/add in the new model. Note: the data is not validated before creating the new model. You should trust this data.
deep (bool) – Set to True to make a deep copy of the model.

Returns:

New model instance.

Return type:

typing_extensions.Self

model_dump(*, mode='python', include=None, exclude=None, context=None, by_alias=False, exclude_unset=False, exclude_defaults=False, exclude_none=False, round_trip=False, warnings=True, serialize_as_any=False)

Usage docs: https://docs.pydantic.dev/2.9/concepts/serialization/#modelmodel_dump

Generate a dictionary representation of the model, optionally specifying which fields to include or exclude.

Parameters:

mode (Literal['json', 'python'] | str) – The mode in which to_python should run. If mode is ‘json’, the output will only contain JSON serializable types. If mode is ‘python’, the output may contain non-JSON-serializable Python objects.
include (IncEx | None) – A set of fields to include in the output.
exclude (IncEx | None) – A set of fields to exclude from the output.
context (Any | None) – Additional context to pass to the serializer.
by_alias (bool) – Whether to use the field’s alias in the dictionary key if defined.
exclude_unset (bool) – Whether to exclude fields that have not been explicitly set.
exclude_defaults (bool) – Whether to exclude fields that are set to their default value.
exclude_none (bool) – Whether to exclude fields that have a value of None.
round_trip (bool) – If True, dumped values should be valid as input for non-idempotent types such as Json[T].
warnings (bool | Literal['none', 'warn', 'error']) – How to handle serialization errors. False/”none” ignores them, True/”warn” logs errors, “error” raises a [PydanticSerializationError][pydantic_core.PydanticSerializationError].
serialize_as_any (bool) – Whether to serialize fields with duck-typing serialization behavior.

Returns:

A dictionary representation of the model.

Return type:

dict[str, Any]

model_dump_json(*, indent=None, include=None, exclude=None, context=None, by_alias=False, exclude_unset=False, exclude_defaults=False, exclude_none=False, round_trip=False, warnings=True, serialize_as_any=False)

Usage docs: https://docs.pydantic.dev/2.9/concepts/serialization/#modelmodel_dump_json

Generates a JSON representation of the model using Pydantic’s to_json method.

Parameters:

indent (int | None) – Indentation to use in the JSON output. If None is passed, the output will be compact.
include (IncEx | None) – Field(s) to include in the JSON output.
exclude (IncEx | None) – Field(s) to exclude from the JSON output.
context (Any | None) – Additional context to pass to the serializer.
by_alias (bool) – Whether to serialize using field aliases.
exclude_unset (bool) – Whether to exclude fields that have not been explicitly set.
exclude_defaults (bool) – Whether to exclude fields that are set to their default value.
exclude_none (bool) – Whether to exclude fields that have a value of None.
round_trip (bool) – If True, dumped values should be valid as input for non-idempotent types such as Json[T].
warnings (bool | Literal['none', 'warn', 'error']) – How to handle serialization errors. False/”none” ignores them, True/”warn” logs errors, “error” raises a [PydanticSerializationError][pydantic_core.PydanticSerializationError].
serialize_as_any (bool) – Whether to serialize fields with duck-typing serialization behavior.

Returns:

A JSON string representation of the model.

Return type:

str

classmethod model_json_schema(by_alias=True, ref_template=DEFAULT_REF_TEMPLATE, schema_generator=GenerateJsonSchema, mode='validation')

Generates a JSON schema for a model class.

Parameters:

by_alias (bool) – Whether to use attribute aliases or not.
ref_template (str) – The reference template.
schema_generator (type[pydantic.json_schema.GenerateJsonSchema]) – To override the logic used to generate the JSON schema, as a subclass of GenerateJsonSchema with your desired modifications
mode (pydantic.json_schema.JsonSchemaMode) – The mode in which to generate the schema.

Returns:

The JSON schema for the given model class.

Return type:

dict[str, Any]

classmethod model_parametrized_name(params)

Compute the class name for parametrizations of generic classes.

This method can be overridden to achieve a custom naming scheme for generic BaseModels.

Parameters:: params (tuple[type[Any], Ellipsis]) – Tuple of types of the class. Given a generic class Model with 2 type variables and a concrete model Model[str, int], the value (str, int) would be passed to params.
Returns:: String representing the new class where `params` are passed to `cls` as type variables.
Raises:: TypeError – Raised when trying to generate concrete names for non-generic models.
Return type:: str

model_post_init(__context)

Override this method to perform additional initialization after __init__ and model_construct. This is useful if you want to do some validation that requires the entire model to be initialized.

Parameters:: __context (Any)
Return type:: None

classmethod model_rebuild(*, force=False, raise_errors=True, _parent_namespace_depth=2, _types_namespace=None)

Try to rebuild the pydantic-core schema for the model.

This may be necessary when one of the annotations is a ForwardRef which could not be resolved during the initial attempt to build the schema, and automatic rebuilding fails.

Parameters:

force (bool) – Whether to force the rebuilding of the model schema, defaults to False.
raise_errors (bool) – Whether to raise errors, defaults to True.
_parent_namespace_depth (int) – The depth level of the parent namespace, defaults to 2.
_types_namespace (dict[str, Any] | None) – The types namespace, defaults to None.

Returns:

Returns `None` if the schema is already “complete” and rebuilding was not required.
If rebuilding _was_ required, returns `True` if rebuilding was successful, otherwise `False`.

Return type:

bool | None

classmethod model_validate(obj, *, strict=None, from_attributes=None, context=None)

Validate a pydantic model instance.

Parameters:

obj (Any) – The object to validate.
strict (bool | None) – Whether to enforce types strictly.
from_attributes (bool | None) – Whether to extract data from object attributes.
context (Any | None) – Additional context to pass to the validator.

Raises:

ValidationError – If the object could not be validated.

Returns:

The validated model instance.

Return type:

typing_extensions.Self

classmethod model_validate_json(json_data, *, strict=None, context=None)

Usage docs: https://docs.pydantic.dev/2.9/concepts/json/#json-parsing

Validate the given JSON data against the Pydantic model.

Parameters:

json_data (str | bytes | bytearray) – The JSON data to validate.
strict (bool | None) – Whether to enforce types strictly.
context (Any | None) – Extra variables to pass to the validator.

Returns:

The validated Pydantic model.

Raises:

ValidationError – If json_data is not a JSON string or the object could not be validated.

Return type:

typing_extensions.Self

classmethod model_validate_strings(obj, *, strict=None, context=None)

Validate the given object with string data against the Pydantic model.

Parameters:

obj (Any) – The object containing string data to validate.
strict (bool | None) – Whether to enforce types strictly.
context (Any | None) – Extra variables to pass to the validator.

Returns:

The validated Pydantic model.

Return type:

typing_extensions.Self

classmethod __get_pydantic_core_schema__(source, handler, /)

Hook into generating the model’s CoreSchema.

Parameters:

source (type[BaseModel]) – The class we are generating a schema for. This will generally be the same as the cls argument if this is a classmethod.
handler (pydantic.annotated_handlers.GetCoreSchemaHandler) – A callable that calls into Pydantic’s internal CoreSchema generation logic.

Returns:

A `pydantic-core` `CoreSchema`.

Return type:

pydantic_core.CoreSchema

classmethod __get_pydantic_json_schema__(core_schema, handler, /)

Hook into generating the model’s JSON schema.

Parameters:

core_schema (pydantic_core.CoreSchema) – A pydantic-core CoreSchema. You can ignore this argument and call the handler with a new CoreSchema, wrap this CoreSchema ({‘type’: ‘nullable’, ‘schema’: current_schema}), or just call the handler with the original schema.
handler (pydantic.annotated_handlers.GetJsonSchemaHandler) – Call into Pydantic’s internal JSON schema generation. This will raise a pydantic.errors.PydanticInvalidForJsonSchema if JSON schema generation fails. Since this gets called by BaseModel.model_json_schema you can override the schema_generator argument to that function to change JSON schema generation globally for a type.

Returns:

A JSON schema, as a Python object.

Return type:

pydantic.json_schema.JsonSchemaValue

classmethod __pydantic_init_subclass__(**kwargs)

This is intended to behave just like __init_subclass__, but is called by ModelMetaclass only after the class is actually fully initialized. In particular, attributes like model_fields will be present when this is called.

This is necessary because __init_subclass__ will always be called by type.__new__, and it would require a prohibitively large refactor to the ModelMetaclass to ensure that type.__new__ was called in such a manner that the class would already be sufficiently initialized.

This will receive the same kwargs that would be passed to the standard __init_subclass__, namely, any kwargs passed to the class definition that aren’t used internally by pydantic.

Parameters:: **kwargs (Any) – Any keyword arguments passed to the class definition that aren’t used internally by pydantic.
Return type:: None

classmethod __class_getitem__(typevar_values)

Parameters:: typevar_values (type[Any] | tuple[type[Any], Ellipsis])
Return type:: type[BaseModel] | pydantic._internal._forward_ref.PydanticRecursiveRef

__copy__()

Returns a shallow copy of the model.

Return type:: typing_extensions.Self

__deepcopy__(memo=None)

Returns a deep copy of the model.

Parameters:: memo (dict[int, Any] | None)
Return type:: typing_extensions.Self

__getattr__(item)

Parameters:: item (str)
Return type:: Any

_check_frozen(name, value)

Parameters:

name (str)
value (Any)

Return type:

None

__getstate__()

Return type:: dict[Any, Any]

__setstate__(state)

Parameters:: state (dict[Any, Any])
Return type:: None

__eq__(other)

Parameters:: other (Any)
Return type:: bool

classmethod __init_subclass__(**kwargs)

This signature is included purely to help type-checkers check arguments to class declaration, which provides a way to conveniently set model_config key/value pairs.

```py from pydantic import BaseModel

class MyModel(BaseModel, extra=’allow’): … ```

However, this may be deceiving, since the _actual_ calls to __init_subclass__ will not receive any of the config arguments, and will only receive any keyword arguments passed during class initialization that are _not_ expected keys in ConfigDict. (This is due to the way ModelMetaclass.__new__ works.)

Parameters:: **kwargs (typing_extensions.Unpack[pydantic.config.ConfigDict]) – Keyword arguments passed to the class definition, which set model_config

Note

You may want to override __pydantic_init_subclass__ instead, which behaves similarly but is called after the class is fully initialized.

__iter__()

So dict(model) works.

Return type:: TupleGenerator

__repr__()

Return type:: str

__repr_args__()

Return type:: pydantic._internal._repr.ReprArgs

__repr_name__

__repr_str__

__pretty__

__rich_repr__

__str__()

Return type:: str

property __fields__: dict[str, pydantic.fields.FieldInfo]

Return type:: dict[str, pydantic.fields.FieldInfo]

property __fields_set__: set[str]

Return type:: set[str]

dict(*, include=None, exclude=None, by_alias=False, exclude_unset=False, exclude_defaults=False, exclude_none=False)

Parameters:

include (IncEx | None)
exclude (IncEx | None)
by_alias (bool)
exclude_unset (bool)
exclude_defaults (bool)
exclude_none (bool)

Return type:

Dict[str, Any]

json(*, include=None, exclude=None, by_alias=False, exclude_unset=False, exclude_defaults=False, exclude_none=False, encoder=PydanticUndefined, models_as_dict=PydanticUndefined, **dumps_kwargs)

Parameters:

include (IncEx | None)
exclude (IncEx | None)
by_alias (bool)
exclude_unset (bool)
exclude_defaults (bool)
exclude_none (bool)
encoder (Callable[[Any], Any] | None)
models_as_dict (bool)
dumps_kwargs (Any)

Return type:

str

classmethod parse_obj(obj)

Parameters:: obj (Any)
Return type:: typing_extensions.Self

classmethod parse_raw(b, *, content_type=None, encoding='utf8', proto=None, allow_pickle=False)

Parameters:

b (str | bytes)
content_type (str | None)
encoding (str)
proto (pydantic.deprecated.parse.Protocol | None)
allow_pickle (bool)

Return type:

typing_extensions.Self

classmethod parse_file(path, *, content_type=None, encoding='utf8', proto=None, allow_pickle=False)

Parameters:

path (str | pathlib.Path)
content_type (str | None)
encoding (str)
proto (pydantic.deprecated.parse.Protocol | None)
allow_pickle (bool)

Return type:

typing_extensions.Self

classmethod from_orm(obj)

Parameters:: obj (Any)
Return type:: typing_extensions.Self

classmethod construct(_fields_set=None, **values)

Parameters:

_fields_set (set[str] | None)
values (Any)

Return type:

typing_extensions.Self

copy(*, include=None, exclude=None, update=None, deep=False)

Returns a copy of the model.

!!! warning “Deprecated”: This method is now deprecated; use model_copy instead.

If you need include or exclude, use:

`py data = self.model_dump(include=include, exclude=exclude, round_trip=True) data = {**data, **(update or {})} copied = self.model_validate(data) `

Parameters:

include (pydantic._internal._utils.AbstractSetIntStr | pydantic._internal._utils.MappingIntStrAny | None) – Optional set or mapping specifying which fields to include in the copied model.
exclude (pydantic._internal._utils.AbstractSetIntStr | pydantic._internal._utils.MappingIntStrAny | None) – Optional set or mapping specifying which fields to exclude in the copied model.
update (Dict[str, Any] | None) – Optional dictionary of field-value pairs to override field values in the copied model.
deep (bool) – If True, the values of fields that are Pydantic models will be deep-copied.

Returns:

A copy of the model with included, excluded and updated fields as specified.

Return type:

typing_extensions.Self

classmethod schema(by_alias=True, ref_template=DEFAULT_REF_TEMPLATE)

Parameters:

by_alias (bool)
ref_template (str)

Return type:

Dict[str, Any]

classmethod schema_json(*, by_alias=True, ref_template=DEFAULT_REF_TEMPLATE, **dumps_kwargs)

Parameters:

by_alias (bool)
ref_template (str)
dumps_kwargs (Any)

Return type:

str

classmethod validate(value)

Parameters:: value (Any)
Return type:: typing_extensions.Self

classmethod update_forward_refs(**localns)

Parameters:: localns (Any)
Return type:: None

_iter(*args, **kwargs)

Parameters:

args (Any)
kwargs (Any)

Return type:

Any

_copy_and_set_values(*args, **kwargs)

Parameters:

args (Any)
kwargs (Any)

Return type:

Any

classmethod _get_value(*args, **kwargs)

Parameters:

args (Any)
kwargs (Any)

Return type:

Any

_calculate_keys(*args, **kwargs)

Parameters:

args (Any)
kwargs (Any)

Return type:

Any

medcat.components.addons.relation_extraction.rel_cat.deserialise(folder_path, ignore_folders_prefix=set(), ignore_folders_suffix=set(), **init_kwargs)

Deserialise contents of a folder.

Extra init keyword arguments can be provided if needed. These are generally: - cnf: The config relevant to the components - tokenizer (BaseTokenizer): The base tokenizer for the model - cdb (CDB): The CDB for the model - vocab (Vocab): The Vocab for the model - model_load_path (Optional[str]): The model load path,

but not the component load path

This method finds the serialiser to be used based on the files on disk.

Parameters:

folder_path (str) – The folder to serialise.
ignore_folders_prefix (set[str]) – The prefixes of folders to ignore.
ignore_folders_suffix (set[str]) – The suffixes of folders to ignore.

Returns:

Serialisable – The deserialised object.

Return type:

medcat.storage.serialisables.Serialisable

class medcat.components.addons.relation_extraction.rel_cat.SerialisingStrategy

Bases: enum.Enum

Describes the strategy for serialising.

SERIALISABLE_ONLY: Only serialise attributes that are of Serialisable type

SERIALISABLES_AND_DICT: Serialise attributes that are Serialisable as well as the rest of .__dict__

DICT_ONLY: Only include the object’s .__dict__

MANUAL

Use manual serialisation defined by the object itself.

NOTE: In this case, most of the logic defined within here will: likely be ignored.

_is_suitable_in_dict(attr_name, attr, obj)

Parameters:

attr_name (str)
attr (Any)
obj (Serialisable)

Return type:

bool

_is_suitable_part(attr_name, part, obj)

Parameters:

attr_name (str)
part (Any)
obj (Serialisable)

Return type:

bool

_iter_obj_items(obj)

Parameters:: obj (Serialisable)
Return type:: Iterable[tuple[str, Any]]

_iter_obj_values(obj)

Parameters:: obj (Serialisable)
Return type:: Iterable[Any]

get_dict(obj)

Gets the appropriate parts of the __dict__ of the object.

I.e this filters out parts that shouldn’t be included.

Parameters:: obj (Serialisable) – The serialisable object.
Returns:: dict[str, Any] – The filtered attributes map.
Return type:: dict[str, Any]

get_parts(obj)

Gets the matching serialisable parts of the object.

This includes only serialisable parts, and only if specified by the strategy.

Returns:: list[tuple[Serialisable, str]] – The serialisable parts with names.
Parameters:: obj (Serialisable)
Return type:: list[tuple[Serialisable, str]]

__new__(value)

_generate_next_value_(start, count, last_values)

Generate the next value when not given.

name: the name of the member start: the initial start value or None count: the number of existing members last_value: the last value assigned or None

classmethod _missing_(value)

__repr__()

__str__()

__dir__(): Returns all members and all public methods

__format__(format_spec): Returns format using actual value type unless __str__ has been overridden.

__hash__()

__reduce_ex__(proto)

name(): The name of the Enum member.

value(): The value of the Enum member.

class medcat.components.addons.relation_extraction.rel_cat.AddonComponent

Bases: medcat.components.types.BaseComponent, Protocol

Base/abstract addon component class.

NAME_PREFIX: str = 'addon_'

NAME_SPLITTER: str = '.'

config: medcat.config.config.ComponentConfig

property addon_type: str

Return type:: str

is_core()

Whether the component is a core component or not.

Returns:: bool – Whether this is a core component.
Return type:: bool

classmethod get_folder_name_for_addon_and_name(addon_type, name)

Parameters:

addon_type (str)
name (str)

Return type:

str

get_folder_name()

Return type:: str

property full_name: str

Name with the component type (e.g ner, linking, meta).

Return type:: str

property include_in_output: bool

Return type:: bool

get_output_key_val(ent)

Parameters:: ent (medcat.components.types.MutableEntity)
Return type:: tuple[str, dict[str, Any]]

property name: str

The name of the component.

Return type:: str

__call__(doc)

Parameters:: doc (medcat.tokenizing.tokens.MutableDocument)
Return type:: medcat.tokenizing.tokens.MutableDocument

classmethod create_new_component(cnf, tokenizer, cdb, vocab, model_load_path)

Create a new component or load one off disk if load path presented.

This may raise an exception if the wrong type of config is provided.

Parameters:

cnf (ComponentConfig) – The config relevant to this components.
tokenizer (BaseTokenizer) – The base tokenizer.
cdb (CDB) – The CDB.
vocab (Vocab) – The Vocab.
model_load_path (Optional[str]) – Model load path (if present).

Returns:

Self – The new components.

Return type:

typing_extensions.Self

__slots__ = ()

_is_protocol = True

_is_runtime_protocol = False

classmethod __init_subclass__(*args, **kwargs)

classmethod __class_getitem__(params)

class medcat.components.addons.relation_extraction.rel_cat.RelExtrBaseComponent(tokenizer=BaseTokenizerWrapper(), model=None, model_config=None, config=ConfigRelCAT(), task='train', init_model=False)

Parameters:

tokenizer (medcat.components.addons.relation_extraction.tokenizer.BaseTokenizerWrapper)
model (medcat.components.addons.relation_extraction.models.RelExtrBaseModel)
model_config (medcat.components.addons.relation_extraction.config.RelExtrBaseConfig)
config (medcat.config.config_rel_cat.ConfigRelCAT)
task (str)
init_model (bool)

name = 'base_component_rel'

__init__(tokenizer=BaseTokenizerWrapper(), model=None, model_config=None, config=ConfigRelCAT(), task='train', init_model=False)

Component that holds the model and everything for RelCAT.

Parameters:

tokenizer (BaseTokenizerWrapper) – The base tokenizer for RelCAT.
model (RelExtrBaseModel) – The model wrapper.
model_config (RelExtrBaseConfig) – The model-specific config.
config (ConfigRelCAT) – The RelCAT config.
task (str) – The task - used for checkpointing.
init_model (bool) – Loads default BERT base model, tokenizer, model config. Defaults to False.

model: medcat.components.addons.relation_extraction.models.RelExtrBaseModel = None

tokenizer: medcat.components.addons.relation_extraction.tokenizer.BaseTokenizerWrapper

relcat_config: medcat.config.config_rel_cat.ConfigRelCAT

model_config: medcat.components.addons.relation_extraction.config.RelExtrBaseConfig = None

optimizer: torch.optim.AdamW = None

scheduler: torch.optim.lr_scheduler.MultiStepLR = None

task: str = 'train'

epoch: int = 0

best_f1: float = 0.0

pad_id

padding_seq

save(save_path)

Saves model and its dependencies to specified save_path folder.: The CDB is obviously not saved, it is however necessary to save the tokenizer used.

Parameters:: save_path (str) – folder path in which to save the model & deps.
Return type:: None

classmethod load(pretrained_model_name_or_path='./')

Parameters:: pretrained_model_name_or_path (str) – Path to RelCAT model. Defaults to “./”.
Returns:: RelExtrBaseComponent – component.
Return type:: RelExtrBaseComponent

classmethod from_relcat_config(relcat_config, pretrained_model_name_or_path='./')

Parameters:

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

Return type:

RelExtrBaseComponent

medcat.components.addons.relation_extraction.rel_cat.set_all_seeds(seed)

Parameters:: seed (int)
Return type:: None

medcat.components.addons.relation_extraction.rel_cat.load_results(path, model_name='BERT', file_prefix='train')

Parameters:

path (str)
model_name (str)
file_prefix (str)

Return type:

tuple[list, list, list]

medcat.components.addons.relation_extraction.rel_cat.load_state(model, optimizer, scheduler, path='./', model_name='BERT', file_prefix='train', load_best=False, relcat_config=ConfigRelCAT())

Used by RelCAT.load() and RelCAT.train()

Parameters:

model (RelExtrBaseModel) – RelExtrBaseModel, it has to be initialized before calling this method via RelExtr(Bert/Llama)Model(…)
optimizer (_type_) – optimizer
scheduler (_type_) – scheduler
path (str, optional) – Defaults to “./”.
model_name (str, optional) – Defaults to “BERT”.
file_prefix (str, optional) – Defaults to “train”.
load_best (bool, optional) – Defaults to False.
relcat_config (ConfigRelCAT) – Defaults to ConfigRelCAT().

Returns:

tuple (int, int) – last epoch and f1 score.

Return type:

tuple[int, int]

medcat.components.addons.relation_extraction.rel_cat.save_results(data, model_name='BERT', path='./', file_prefix='train')

Parameters:

model_name (str)
path (str)
file_prefix (str)

medcat.components.addons.relation_extraction.rel_cat.save_state(model, optimizer, scheduler, epoch=1, best_f1=0.0, path='./', model_name='BERT', task='train', is_checkpoint=False, final_export=False)

Used by RelCAT.save() and RelCAT.train(): Saves the RelCAT model state. For checkpointing multiple files are created, best_f1, loss etc. score. If you want to export the model after training set final_export=True and leave is_checkpoint=False.

Parameters:

model (BaseModel) – BertMode | LlamaModel etc.
optimizer (torch.optim.AdamW, optional) – Defaults to None.
scheduler (torch.optim.lr_scheduler.MultiStepLR, optional) – Defaults to None.
epoch (int) – Defaults to None.
best_f1 (float) – Defaults to None.
path (str) – Defaults to “./”.
model_name (str) – . Defaults to “BERT”. This is used to checkpointing only.
task (str) – Defaults to “train”. This is used to checkpointing only.
is_checkpoint (bool) – Defaults to False.
final_export (bool) – Defaults to False, if True then is_checkpoint must be False also. Exports model.state_dict(), out into “model.dat”.

Return type:

None

medcat.components.addons.relation_extraction.rel_cat.split_list_train_test_by_class(data, sample_limit=-1, test_size=0.2, shuffle=True)

Parameters:

data (list) – “output_relations”: relation_instances, see create_base_relations_from_doc/csv for data columns
sample_limit (int) – Limit the number of samples per class, useful for dataset balancing . Defaults to -1.
test_size (float) – Defaults to 0.2.
shuffle (bool) – Shuffle data randomly. Defaults to True.

Returns:

tuple[list, list] – Train and test datasets

Return type:

tuple[list, list]

class medcat.components.addons.relation_extraction.rel_cat.RelData(tokenizer, config, cdb=CDB(CoreConfig()))

Bases: torch.utils.data.Dataset

An abstract class representing a Dataset.

All datasets that represent a map from keys to data samples should subclass it. All subclasses should overwrite __getitem__(), supporting fetching a data sample for a given key. Subclasses could also optionally overwrite __len__(), which is expected to return the size of the dataset by many Sampler implementations and the default options of DataLoader. Subclasses could also optionally implement __getitems__(), for speedup batched samples loading. This method accepts list of indices of samples of batch and returns list of samples.

Note

DataLoader by default constructs an index sampler that yields integral indices. To make it work with a map-style dataset with non-integral indices/keys, a custom sampler must be provided.

Parameters:

tokenizer (medcat.components.addons.relation_extraction.tokenizer.BaseTokenizerWrapper)
config (medcat.config.config_rel_cat.ConfigRelCAT)
cdb (medcat.cdb.CDB)

name = 'rel_dataset'

__init__(tokenizer, config, cdb=CDB(CoreConfig()))

Use this class to create a dataset for relation annotations from CSV exports, MedCAT exports or Spacy Documents (assuming the documents got generated by MedCAT, if they did not then please set the required parameters manually to match MedCAT output, see /medcat/cat.py#_get_entity)

If you are using this to create relations from CSV it is assumed that your entities/concepts of interest are surrounded by the special tokens, see create_base_relations_from_csv doc.

Parameters:

tokenizer (BaseTokenizerWrapper) – tokenizer used to generate token ids from input text
config (ConfigRelCAT) – same config used in RelCAT
cdb (CDB) – Optional, used to add concept ids and types to detected ents, useful when creating datasets from MedCAT output. Defaults to CDB().

cdb: medcat.cdb.CDB

config: medcat.config.config_rel_cat.ConfigRelCAT

tokenizer: medcat.components.addons.relation_extraction.tokenizer.BaseTokenizerWrapper

dataset: dict[Any, Any]

generate_base_relations(docs)

Util function, should be used if you want to train from spacy docs

Parameters:

docs (Iterable[MutableDocument]) – Generate relations from Spacy CAT docs.

Returns:

output_relations – list[dict] : [] “output_relations”: relation_instances, # NOTE: see create_base_relations_from_doc/csv

for data columns

“nclasses”: self.config.model.padding_idx # dummy class “labels2idx”: {}, “idx2label”: {}} ]

Return type:

list[dict]

create_base_relations_from_csv(csv_path, keep_source_text=False)

Assumes the columns are as follows [“relation_token_span_ids”,

“ent1_ent2_start”, “ent1”, “ent2”, “label”, “label_id”, “ent1_type”, “ent2_type”, “ent1_id”, “ent2_id”, “ent1_cui”, “ent2_cui”, “doc_id”, “sents”],

last column is the actual source text.

The entities inside the text MUST be annotated with special tokens i.e:

…text..[s1] first ent [e1]…..[s2] second ent [e2]……..

You have to store the start position, aka index position of token [e1] and also of token [e2] in the (ent1_ent2_start) column.

Parameters:

csv_path (str) – Path to csv file, must have specific columns, tab separated.
keep_source_text (bool) – If the text clumn should be retained in the ‘sents’ df column, used for debugging or creating custom datasets.

Returns:

dict – { “output_relations”: relation_instances, # NOTE: see create_base_relations_from_doc/csv

for data columns

“nclasses”: self.config.model.padding_idx, # dummy class “labels2idx”: {}, “idx2label”: {}}
}

_create_relation_validation(text, doc_id, tokenized_text_data, ent1_start_char_pos, ent2_start_char_pos, ent1_end_char_pos, ent2_end_char_pos, ent1_token_start_pos=-1, ent2_token_start_pos=-1, ent1_token_end_pos=-1, ent2_token_end_pos=-1, is_spacy_doc=False, is_mct_export=False)

This function checks if the relation is actually valid by distance criteria, TUIs and so on. Has diffierent handling cases for text, spacy docs and MCT exports.

Parameters:

text (str) – doc text
doc_id (str) – doc id
tokenized_text_data (dict[str, Any]) – tokenized text
ent1_start_char_pos (int) – ent1 start char pos
ent2_start_char_pos (int) – ent2 start char pos
ent1_end_char_pos (int) – ent1 end char pos
ent2_end_char_pos (int) – ent2 end char pos
ent1_token_start_pos (int) – ent1_token_start_pos. Defaults to -1.
ent2_token_start_pos (int) – ent2_token_start_pos. Defaults to -1.
ent1_token_end_pos (int) – ent1_token_end_pos. Defaults to -1.
ent2_token_end_pos (int) – ent2_token_end_pos. Defaults to -1.
is_spacy_doc (bool) – checks if doc is spacy docs. Defaults to False.
is_mct_export (bool) – chekcs if doc is a mct export. Defaults to False.

Returns:

list – row containing rel data [“relation_token_span_ids”, “ent1_ent2_start”, “ent1”, “ent2”, “label”, “label_id”, “ent1_type”, “ent2_type”, “ent1_id”, “ent2_id”, “ent1_cui”, “ent2_cui”, “doc_id”, “sents”]

Return type:

list

_get_token_type_and_start_end(entity, doc_length_tokens, tokenized_text_data)

Parameters:

entity (medcat.tokenizing.tokens.MutableEntity)
doc_length_tokens (int)
tokenized_text_data (dict[str, Any])

Return type:

tuple[list[str], tuple[int, int], tuple[int, int]]

_create_base_relation_for_ents(doc_text, doc_id, ent1_token, ent2_token, tokenized_text_data, chars_to_exclude, doc_length_tokens)

Parameters:

doc_text (str)
doc_id (str)
ent1_token (medcat.tokenizing.tokens.MutableEntity)
ent2_token (medcat.tokenizing.tokens.MutableEntity)
tokenized_text_data (dict[str, Any])
chars_to_exclude (str)
doc_length_tokens (int)

Return type:

Optional[list]

_create_base_relations_from_mutable_doc(doc, doc_text, doc_id, tokenized_text_data, doc_length_tokens, chars_to_exclude)

Parameters:

doc (medcat.tokenizing.tokens.MutableDocument)
doc_text (str)
doc_id (str)
tokenized_text_data (dict[str, Any])
doc_length_tokens (int)
chars_to_exclude (str)

Return type:

list[list]

create_base_relations_from_doc(doc, doc_id, ent1_ent2_tokens_start_pos=(-1, -1))

Creates a list of tuples based on pairs of entities detected (relation, ent1, ent2) for one spacy document or text string.

Parameters:

doc (Union[MutableDocument, str]) – SpacyDoc or string of text, each will get handled slightly differently
doc_id (str) – Document id
ent1_ent2_tokens_start_pos (Union[list, tuple], optional) – Start of [s1][s2] tokens, if left default we assume we are dealing with a SpacyDoc. Defaults to (-1, -1).

Returns:

dict – { # NOTE: see create_base_relations_from_doc/csv # for data columns “output_relations”: relation_instances, “nclasses”: self.config.model.padding_idx, # dummy class “labels2idx”: {}, “idx2label”: {}}
}

Return type:

dict

_create_relations_for_doc(document, data)

Parameters:

document (medcat.data.mctexport.MedCATTrainerExportDocument)
data (dict)

Return type:

list[list]

create_relations_from_export(data)

Parameters:

data (dict) – MedCAT Export data.

Returns:

dict – { # NOTE: see create_base_relations_from_doc/csv

for data columns

“output_relations”: relation_instances, “nclasses”: self.config.model.padding_idx, # dummy class “labels2idx”: {}, “idx2label”: {}}
}

classmethod get_labels(relation_labels, config)

This is used to update labels in config with unencountered classes/labels ( if any are encountered during training).

Parameters:

relation_labels (list[str]) – new labels to add
config (ConfigRelCAT) – config

Returns:

tuple[int, dict[str, int], dict[int, str]] – label count, labesl2idx mapping, idx2labels mapping

Return type:

tuple[int, dict[str, int], dict[int, str]]

__len__()

Returns:: int – num of rels records
Return type:: int

__getitem__(idx)

Parameters:

idx (int) – index of item in the dataset dict

Returns:

tuple[torch.LongTensor, torch.LongTensor, torch.LongTensor] –

Long tensors of the following the columns :: input_ids, ent1&ent2 token start pos idx, label_ids

Return type:

tuple[torch.LongTensor, torch.LongTensor, torch.LongTensor]

__add__(other)

Parameters:: other (Dataset[_T_co])
Return type:: ConcatDataset[_T_co]

__slots__ = ()

_is_protocol = False

classmethod __class_getitem__(params)

classmethod __init_subclass__(*args, **kwargs)

class medcat.components.addons.relation_extraction.rel_cat.BaseTokenizer

Bases: Protocol

The base tokenizer protocol.

create_entity(doc, token_start_index, token_end_index, label)

Create an entity from a document.

Parameters:

doc (MutableDocument) – The document to use.
token_start_index (int) – The token start index.
token_end_index (int) – The token end index.
label (str) – The label.

Returns:

MutableEntity – The resulting entity.

Return type:

medcat.tokenizing.tokens.MutableEntity

entity_from_tokens(tokens)

Get an entity from the list of tokens.

Parameters:: tokens (list[MutableToken]) – List of tokens.
Returns:: MutableEntity – The resulting entity.
Return type:: medcat.tokenizing.tokens.MutableEntity

__call__(text)

Parameters:: text (str)
Return type:: medcat.tokenizing.tokens.MutableDocument

classmethod create_new_tokenizer(config)

Parameters:: config (medcat.config.Config)
Return type:: typing_extensions.Self

get_doc_class()

Get the document implementation class used by the tokenizer.

This can be used (e.g) to register addon paths.

Returns:: Type[MutableDocument] – The document class.
Return type:: Type[medcat.tokenizing.tokens.MutableDocument]

get_entity_class()

Get the entity implementation class used by the tokenizer.

Returns:: Type[MutableEntity] – The entity class.
Return type:: Type[medcat.tokenizing.tokens.MutableEntity]

__slots__ = ()

_is_protocol = True

_is_runtime_protocol = False

classmethod __init_subclass__(*args, **kwargs)

classmethod __class_getitem__(params)

medcat.components.addons.relation_extraction.rel_cat.create_tokenizer(tokenizer_name, config)

Create the tokenizer given the init arguments.

Parameters:

tokenizer_name (str) – The tokenizer name.
config (Config) – The config to be passed to the constructor.

Returns:

BaseTokenizer – The created tokenizer.

Return type:

BaseTokenizer

class medcat.components.addons.relation_extraction.rel_cat.MutableDocument

Bases: Protocol

The mutable parts of the document.

Represents parts of the document that can / should be changed by the various components.

property base: BaseDocument

The base document.

Return type:: BaseDocument

property linked_ents: list[MutableEntity]

The linked entities associated with the document.

This should be set by the linker.

Return type:: list[MutableEntity]

property ner_ents: list[MutableEntity]

All entities recognised by NER.

This should be set by the NER component.

Return type:: list[MutableEntity]

__iter__()

Return type:: Iterator[MutableToken]

__getitem__(index: int) → MutableToken
__getitem__(index: slice) → MutableEntity

__len__()

Return type:: int

get_tokens(start_index, end_index)

Get the tokens that span the specified character indices.

Parameters:

start_index (int) – The starting character index.
end_index (int) – The ending character index.

Returns:

list[MutableToken] – The list of tokens.

Return type:

list[MutableToken]

set_addon_data(path, val)

Used to add arbitrary data to the entity.

This is generally used by addons to keep track of their data.

NB! The path used needs to be registered using the register_addon_path class method.

Parameters:

path (str) – The data ID / path.
val (Any) – The value to be added.

Return type:

None

has_addon_data(path)

Checks whether the addon data for a specific path has been set.

Parameters:: path (str) – The path to check.
Returns:: bool – Whether the addon data had been set.
Return type:: bool

get_addon_data(path)

Get data added to the entity.

See add_data for details.

Parameters:: path (str) – The data ID / path.
Returns:: Any – The stored value.
Return type:: Any

get_available_addon_paths()

Gets the available addon data paths for this document.

This will only include paths that have values set.

Returns:: list[str] – List of available addon data paths.
Return type:: list[str]

classmethod register_addon_path(path, def_val=None, force=True)

Register a custom/arbitrary data path.

This can be used to store arbitrary data along with the entity for use in an addon (e.g MetaCAT).

PS: If using this, it is important to use paths namespaced to the component you’re using in order to avoid conflicts.

Parameters:

path (str) – The path to be used. Should be prefixed by component name (e.g meta_cat_id for an ID tied to the meta_cat addon)
def_val (Any) – Default value. Defaults to None.
force (bool) – Whether to forcefully add the value. Defaults to True.

Return type:

None

__slots__ = ()

_is_protocol = True

_is_runtime_protocol = False

classmethod __init_subclass__(*args, **kwargs)

classmethod __class_getitem__(params)

medcat.components.addons.relation_extraction.rel_cat.COMPONENTS_FOLDER = 'saved_components'

medcat.components.addons.relation_extraction.rel_cat.logger

class medcat.components.addons.relation_extraction.rel_cat.RelCATAddon(config, rel_cat)

Bases: medcat.components.addons.addons.AddonComponent

Base/abstract addon component class.

Parameters:

config (medcat.config.config_rel_cat.ConfigRelCAT)
rel_cat (RelCAT)

addon_type = 'rel_cat'

output_key = 'relations'

config: medcat.config.config_rel_cat.ConfigRelCAT

__init__(config, rel_cat)

Parameters:

config (medcat.config.config_rel_cat.ConfigRelCAT)
rel_cat (RelCAT)

_rel_cat

classmethod create_new(config, base_tokenizer, cdb)

Factory method to create a new MetaCATAddon instance.

Parameters:

config (medcat.config.config_rel_cat.ConfigRelCAT)
base_tokenizer (medcat.tokenizing.tokenizers.BaseTokenizer)
cdb (medcat.cdb.CDB)

Return type:

RelCATAddon

classmethod create_new_component(cnf, tokenizer, cdb, vocab, model_load_path)

Create a new component or load one off disk if load path presented.

This may raise an exception if the wrong type of config is provided.

Parameters:

cnf (ComponentConfig) – The config relevant to this components.
tokenizer (BaseTokenizer) – The base tokenizer.
cdb (CDB) – The CDB.
vocab (Vocab) – The Vocab.
model_load_path (Optional[str]) – Model load path (if present).

Returns:

Self – The new components.

Return type:

RelCATAddon

classmethod load_existing(cnf, base_tokenizer, cdb, load_path)

Factory method to load an existing RelCAT addon from disk.

Parameters:

cnf (medcat.config.config_rel_cat.ConfigRelCAT)
base_tokenizer (medcat.tokenizing.tokenizers.BaseTokenizer)
cdb (medcat.cdb.CDB)
load_path (str)

Return type:

RelCATAddon

serialise_to(folder_path)

Parameters:: folder_path (str)
Return type:: None

property name: str

The name of the component.

Return type:: str

classmethod deserialise_from(folder_path, **init_kwargs)

Parameters:: folder_path (str)
Return type:: RelCATAddon

get_strategy()

Return type:: medcat.storage.serialisables.SerialisingStrategy

classmethod get_init_attrs()

Return type:: list[str]

classmethod ignore_attrs()

Return type:: list[str]

classmethod include_properties()

Return type:: list[str]

__call__(doc)

Parameters:: doc (medcat.tokenizing.tokens.MutableDocument)

NAME_PREFIX: str = 'addon_'

NAME_SPLITTER: str = '.'

is_core()

Whether the component is a core component or not.

Returns:: bool – Whether this is a core component.
Return type:: bool

classmethod get_folder_name_for_addon_and_name(addon_type, name)

Parameters:

addon_type (str)
name (str)

Return type:

str

get_folder_name()

Return type:: str

property full_name: str

Name with the component type (e.g ner, linking, meta).

Return type:: str

property include_in_output: bool

Return type:: bool

get_output_key_val(ent)

Parameters:: ent (medcat.components.types.MutableEntity)
Return type:: tuple[str, dict[str, Any]]

__slots__ = ()

_is_protocol = True

_is_runtime_protocol = False

classmethod __init_subclass__(*args, **kwargs)

classmethod __class_getitem__(params)

class medcat.components.addons.relation_extraction.rel_cat.BalancedBatchSampler(dataset, classes, batch_size, max_samples, max_minority)

Bases: torch.utils.data.Sampler

Base class for all Samplers.

Every Sampler subclass has to provide an __iter__() method, providing a way to iterate over indices or lists of indices (batches) of dataset elements, and may provide a __len__() method that returns the length of the returned iterators.

Parameters:: data_source (Dataset) – This argument is not used and will be removed in 2.2.0. You may still have custom implementation that utilizes it.

Example

>>> # xdoctest: +SKIP
>>> class AccedingSequenceLengthSampler(Sampler[int]):
>>>     def __init__(self, data: List[str]) -> None:
>>>         self.data = data
>>>
>>>     def __len__(self) -> int:
>>>         return len(self.data)
>>>
>>>     def __iter__(self) -> Iterator[int]:
>>>         sizes = torch.tensor([len(x) for x in self.data])
>>>         yield from torch.argsort(sizes).tolist()
>>>
>>> class AccedingSequenceLengthBatchSampler(Sampler[List[int]]):
>>>     def __init__(self, data: List[str], batch_size: int) -> None:
>>>         self.data = data
>>>         self.batch_size = batch_size
>>>
>>>     def __len__(self) -> int:
>>>         return (len(self.data) + self.batch_size - 1) // self.batch_size
>>>
>>>     def __iter__(self) -> Iterator[List[int]]:
>>>         sizes = torch.tensor([len(x) for x in self.data])
>>>         for batch in torch.chunk(torch.argsort(sizes), len(self)):
>>>             yield batch.tolist()

Note

The __len__() method isn’t strictly required by DataLoader, but is expected in any calculation involving the length of a DataLoader.

__init__(dataset, classes, batch_size, max_samples, max_minority)

dataset

classes

batch_size

num_classes

indices

max_minority

max_samples_per_class

__len__()

__iter__()

__slots__ = ()

_is_protocol = False

classmethod __class_getitem__(params)

classmethod __init_subclass__(*args, **kwargs)

class medcat.components.addons.relation_extraction.rel_cat.RelCAT(base_tokenizer, cdb, config=ConfigRelCAT(), task='train', init_model=False)

The RelCAT class used for training ‘Relation-Annotation’ models, i.e., annotation of relations between clinical concepts.

Parameters:

cdb (CDB) – cdb, this is used when creating relation datasets.
tokenizer (TokenizerWrapperBERT) – The Huggingface tokenizer instance. This can be a pre-trained tokenzier instance from a BERT-style model. For now, only BERT models are supported.
config (ConfigRelCAT) – the configuration for RelCAT. Param descriptions available in ConfigRelCAT class docs.
task (str, optional) – What task is this model supposed to handle. Defaults to “train”
init_model (bool, optional) – loads default model. Defaults to False.
base_tokenizer (medcat.tokenizing.tokenizers.BaseTokenizer)

addon_type = 'rel_cat'

output_key = 'rel_'

__init__(base_tokenizer, cdb, config=ConfigRelCAT(), task='train', init_model=False)

Parameters:

base_tokenizer (medcat.tokenizing.tokenizers.BaseTokenizer)
cdb (medcat.cdb.CDB)
config (medcat.config.config_rel_cat.ConfigRelCAT)
task (str)
init_model (bool)

base_tokenizer

component

task: str = 'train'

checkpoint_path: str = './'

cdb

is_cuda_available = False

device

_init_data_paths()

save(save_path='./')

Parameters:: save_path (str)
Return type:: None

classmethod load(load_path='./')

Parameters:: load_path (str)
Return type:: RelCAT

__call__(doc)

Parameters:: doc (medcat.tokenizing.tokens.MutableDocument)
Return type:: medcat.tokenizing.tokens.MutableDocument

_create_test_train_datasets(data, split_sets=False)

Parameters:

data (dict)
split_sets (bool)

train(export_data_path='', train_csv_path='', test_csv_path='', checkpoint_path='./')

Parameters:

export_data_path (str)
train_csv_path (str)
test_csv_path (str)
checkpoint_path (str)

_train_epoch(epoch, gradient_acc_steps, max_grad_norm, train_dataset_size, train_dataloader, test_dataloader, criterion, _epochs, checkpoint_path)

Parameters:

epoch (int)
gradient_acc_steps (int)
max_grad_norm (float)
train_dataset_size (int)
train_dataloader (torch.utils.data.DataLoader)
test_dataloader (torch.utils.data.DataLoader)
criterion (torch.nn.CrossEntropyLoss)
_epochs (int)
checkpoint_path (str)

Return type:

tuple[list, list, list]

evaluate_(output_logits, labels, ignore_idx)

evaluate_results(data_loader, pad_id)

pipe(stream, *args, **kwargs)

Parameters:: stream (Iterable[medcat.tokenizing.tokens.MutableDocument])
Return type:: Iterator[medcat.tokenizing.tokens.MutableDocument]

predict_text_with_anns(text, annotations)

Creates spacy doc from text and annotation input. Predicts using self.__call__

Parameters:

text (str) – text
annotations (dict) –
dict containing the entities from NER (of your choosing), the format must be the following format:

[

{
“cui”: “202099003”, -this is optional “value”: “discoid lateral meniscus”, “start”: 294, “end”: 318

}, {

”cui”: “202099003”, “value”: “Discoid lateral meniscus”, “start”: 1905, “end”: 1929,

}

]

Returns:

Doc – spacy doc with the relations.

Return type:

medcat.tokenizing.tokens.MutableDocument