Error at Predictor Training: "Predictor is not a subclass of QESystem"

[tervingo]

Hi,

I am trying to train a predictor model with OpenKiwi 2.0.0 on Windows. I am using the command 'kiwi train config\predictor.yaml'. I am getting the error:

Predictor is not a subclass of QESystem; make sure its class is decorated with @QESystem.register_subclass (type=value_error)
system -> load
Must provide class_name or load (type=value_error)

The predictor.yaml file has the line

system:
class_name: Predictor

Any help would be most welcome.

Thanks and Regards

PD - I have tried the same on Ubuntu with OpenKiwi 2.1.0 and I'm getting exactly the same error

[zhaofuchen]

what do you want ? predictor model is a language model, which is the first step of the estimator predictor theory. This model is used to obtain language features, and can not implement QE function

…

------------------ 原始邮件 ------------------ 发件人: "Unbabel/OpenKiwi" ***@***.***>; 发送时间: 2021年4月22日(星期四) 晚上7:26 ***@***.***>; ***@***.***>; 主题: [Unbabel/OpenKiwi] Predictor is not a subclass of QESystem (#97) Hi, I am trying to train a predictor model with OpenKiwi 2.0.0 on Windows. I am using the command 'kiwi train config\predictor.yaml'. I am getting the error: Predictor is not a subclass of QESystem; make sure its class is decorated with @QESystem.register_subclass (type=value_error) system -> load Must provide class_name or load (type=value_error) The predictor.yaml file has the line system: class_name: Predictor Any help would be most welcome. Thanks and Regards — You are receiving this because you are subscribed to this thread. Reply to this email directly, view it on GitHub, or unsubscribe.

[captainvera]

Hello @tervingo,

Sorry for my late response! As @zhaofuchen mentioned, the Predictor is the first step of creating a predictor-estimator model. This means that it can't be trained with the Quality Estimation training pipeline.

Instead, this step is handled by the pretrain pipeline. Doing kiwi pretrain config\predictor.yaml should work!

you can read more at: https://unbabel.github.io/OpenKiwi/usage.html#training-and-pretraining