kiwi.systems.predictor_estimator

Module Contents

Classes

ModelConfig	Base class for all pydantic configs. Used to configure base behaviour of configs.
PredictorEstimator	Predictor-Estimator QE model (proposed in 2017).

kiwi.systems.predictor_estimator.logger

class kiwi.systems.predictor_estimator.ModelConfig

Bases: kiwi.utils.io.BaseConfig

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

encoder :PredictorEncoder.Config

decoder :EstimatorDecoder.Config

outputs :QEOutputs.Config

tlm_outputs :TLMOutputs.Config

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

Bases: kiwi.systems.qe_system.QESystem

Predictor-Estimator QE model (proposed in 2017).

class Config

Bases: kiwi.systems.qe_system.QESystem.Config

System configuration base class.

model :ModelConfig

kiwi.systems.predictor kiwi.systems.qe_system