kiwi.systems.bert

Module Contents

Classes

ModelConfig	Base class for all pydantic configs. Used to configure base behaviour of configs.
Bert	BERT-based Predictor-Estimator model.

kiwi.systems.bert.logger

class kiwi.systems.bert.ModelConfig

Bases: kiwi.utils.io.BaseConfig

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

encoder :BertEncoder.Config

decoder :LinearDecoder.Config

outputs :QEOutputs.Config

tlm_outputs :TLMOutputs.Config

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

Bases: kiwi.systems.qe_system.QESystem

BERT-based Predictor-Estimator model.

class Config

Bases: kiwi.systems.qe_system.QESystem.Config

System configuration base class.

model :ModelConfig

kiwi.systems._meta_module kiwi.systems.nuqe