get_model_cls

get_model_cls(query)[source]

Look up a model class by name (case/punctuation insensitive) in pykeen.models.models.

Parameters

query (Union[str, Type[Model]]) – The name of the model (case insensitive, punctuation insensitive).

Return type

Type[Model]

Returns

The model class