src.wic.deepmistake

`src.wic.deepmistake`#

Module Contents#

Classes#

`Model`
`Cache`
`Score`
`Input`
`DeepMistake`

Functions#

`use_pair_group`(→ str)
`to_data_format`(→ Input)

Attributes#

src.wic.deepmistake.log#

class src.wic.deepmistake.Model#

Bases: pydantic.BaseModel

name: str#

url: pydantic.HttpUrl#

class src.wic.deepmistake.Cache(**data: Any)#

Bases: pydantic.BaseModel

property path: pathlib.Path#

metadata: dict[str, Any]#

_similarities: pandas.DataFrame#

_similarities_filtered: pandas.DataFrame#

__metadata__: pandas.DataFrame#

retrieve(use_pairs: list[tuple[src.use.Use, src.use.Use]]) → dict[tuple[src.use.UseID, src.use.UseID], float]#

persist() → None#

add_use_pair(use_pair: tuple[src.use.Use, src.use.Use], similarity: float) → None#

class src.wic.deepmistake.Score#

Bases: TypedDict

id: str#

score: tuple[str, str] | str#

src.wic.deepmistake.use_pair_group(use_pair: tuple[src.use.Use, src.use.Use]) → str#

class src.wic.deepmistake.Input#

Bases: TypedDict

id: str#

start1: int#

end1: int#

sentence1: str#

start2: int#

end2: int#

sentence2: str#

lemma: str#

pos: str#

grp: str#

src.wic.deepmistake.to_data_format(use_pair: tuple[src.use.Use, src.use.Use]) → Input#

class src.wic.deepmistake.DeepMistake#

Bases: src.wic.model.WICModel

property path: pathlib.Path#

property repo_dir: pathlib.Path#

property ckpt_dir: pathlib.Path#

ckpt: Model#

cache: Cache | None#

__enter__() → None#

__exit__(exc_type, exc_val, exc_tb)#

as_df() → pandas.DataFrame#

clone_repo() → None#

__unzip_ckpt(zipped: pathlib.Path) → None#

__download_ckpt() → pathlib.Path#

predict(use_pairs: list[tuple[src.use.Use, src.use.Use]]) → list[float]#