Closed PastelBelem8 closed 2 years ago
Goal: Implement MOCHA reader and Roberta-base regression model
Main changes
training_config/mocha_eqqa.jsonnet
quality_estimation/mocha_eqqa_reader.py
quality_estimation/mocha_eqqa_roberta_model.py
Goal: Implement MOCHA reader and Roberta-base regression model
Main changes
training_config/mocha_eqqa.jsonnet: Configuration file for the MOCHA-ROBERTA QE experiment.quality_estimation/mocha_eqqa_reader.py: MOCHA dataset reader. This considers the preprocessed version of the files available here.quality_estimation/mocha_eqqa_roberta_model.py: a roberta-base transformer with a regression layer on top.