From c048d7107e0ba2db503cf54360059fcc3350241a Mon Sep 17 00:00:00 2001 From: TharinduDR Date: Sat, 8 May 2021 11:19:00 +0100 Subject: 057: Code Refactoring - Siamese Architectures --- README.md | 2 +- examples/sentence_level/wmt_2018/common/util/draw.py | 2 +- examples/sentence_level/wmt_2020_task2/common/util/draw.py | 6 ++++-- 3 files changed, 6 insertions(+), 4 deletions(-) diff --git a/README.md b/README.md index dcc897c..074b5b8 100644 --- a/README.md +++ b/README.md @@ -8,7 +8,7 @@ With TransQuest, we have opensourced our research in translation quality estimat - Sentence-level translation quality estimation on both aspects: predicting post editing efforts and direct assessment. - Word-level translation quality estimation capable of predicting quality of source words, target words and target gaps. - Perform significantly better than current state-of-the-art quality estimation methods like DeepQuest and OpenKiwi in all the languages experimented. -- Pre-trained quality estimation models for fifteen language pairs. +- Pre-trained quality estimation models for fifteen language pairs are available in [HuggingFace.](https://huggingface.co/TransQuest) ## Table of Contents 1. **[Installation](https://tharindudr.github.io/TransQuest/install/)** - Install TransQuest locally using pip. diff --git a/examples/sentence_level/wmt_2018/common/util/draw.py b/examples/sentence_level/wmt_2018/common/util/draw.py index 3e91099..69ed48c 100644 --- a/examples/sentence_level/wmt_2018/common/util/draw.py +++ b/examples/sentence_level/wmt_2018/common/util/draw.py @@ -2,7 +2,7 @@ import matplotlib.pyplot as plt import pandas as pd from sklearn.metrics import mean_absolute_error -from examples.sentence_level.wmt_2018 import fit +from examples.sentence_level.wmt_2018.common.util.normalizer import fit from transquest.algo.sentence_level.monotransquest.evaluation import pearson_corr, spearman_corr, rmse diff --git a/examples/sentence_level/wmt_2020_task2/common/util/draw.py b/examples/sentence_level/wmt_2020_task2/common/util/draw.py index 9bab846..12c4238 100644 --- a/examples/sentence_level/wmt_2020_task2/common/util/draw.py +++ b/examples/sentence_level/wmt_2020_task2/common/util/draw.py @@ -1,11 +1,13 @@ import pandas as pd from sklearn.metrics import mean_absolute_error -from examples.sentence_level.wmt_2020_task2.common.util import fit -from transquest.algo.monotransquest.evaluation import pearson_corr, spearman_corr, rmse + import matplotlib.pyplot as plt +from examples.sentence_level.wmt_2020_task2.common.util.normalizer import fit +from transquest.algo.sentence_level.monotransquest.evaluation import pearson_corr, spearman_corr, rmse + def draw_scatterplot(data_frame, real_column, prediction_column, path, topic): data_frame = data_frame.sort_values(real_column) -- cgit v1.2.3