kiwi.systems.xlm

Module Contents

Classes

ModelConfig	Base class for all pydantic configs. Used to configure base behaviour of configs.
XLM	XLM-based model for word level quality estimation.

kiwi.systems.xlm.logger

class kiwi.systems.xlm.ModelConfig

Bases: kiwi.utils.io.BaseConfig

Base class for all pydantic configs. Used to configure base behaviour of configs.

encoder :XLMEncoder.Config

decoder :LinearDecoder.Config

outputs :QEOutputs.Config

tlm_outputs :TLMOutputs.Config

class kiwi.systems.xlm.XLM(config, data_config: WMTQEDataset.Config = None, module_dict: Dict[str, Any] = None)

Bases: kiwi.systems.qe_system.QESystem

XLM-based model for word level quality estimation.

class Config

Bases: kiwi.systems.qe_system.QESystem.Config

System configuration base class.

model :ModelConfig

kiwi.systems.tlm_system kiwi.systems.xlmroberta