kiwi.systems.predictor

Module Contents

Classes

ModelConfig

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

Predictor

Predictor TLM, used for the Predictor-Estimator QE model (proposed in 2017).

kiwi.systems.predictor.logger
class kiwi.systems.predictor.ModelConfig

Bases: kiwi.utils.io.BaseConfig

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

encoder :PredictorEncoder.Config
tlm_outputs :TLMOutputs.Config
class kiwi.systems.predictor.Predictor(config: Config, data_config: WMTQEDataset.Config = None, module_dict: Dict[str, Any] = None)

Bases: kiwi.systems.tlm_system.TLMSystem

Predictor TLM, used for the Predictor-Estimator QE model (proposed in 2017).

class Config

Bases: kiwi.systems.tlm_system.TLMSystem.Config

System configuration base class.

model :ModelConfig