diff options
| author | TharinduDR <rhtdranasinghe@gmail.com> | 2021-04-22 18:34:16 +0300 |
|---|---|---|
| committer | TharinduDR <rhtdranasinghe@gmail.com> | 2021-04-22 18:34:16 +0300 |
| commit | 43c091c48a45bb5a2364dd0a92cb57d0e8e3bd03 (patch) | |
| tree | 2c491efaf7b48fad0716b9ea13e221a42003e8d5 | |
| parent | cc6b60648ac4bf59e181446a0e95d5acb4ff72c6 (diff) | |
057: Code Refactoring - Siamese Architectures
| -rw-r--r-- | examples/sentence_level/wmt_2020/common/util/postprocess.py | 3 |
1 files changed, 3 insertions, 0 deletions
diff --git a/examples/sentence_level/wmt_2020/common/util/postprocess.py b/examples/sentence_level/wmt_2020/common/util/postprocess.py index a0782ae..5697909 100644 --- a/examples/sentence_level/wmt_2020/common/util/postprocess.py +++ b/examples/sentence_level/wmt_2020/common/util/postprocess.py @@ -8,6 +8,9 @@ def format_submission(df, language_pair, method, index, path, index_type=None): predictions = df['predictions'].tolist() + print(index) + print(predictions) + with open(path, 'w') as f: for number, prediction in zip(index, predictions): text = language_pair + "\t" + method + "\t" + str(number) + "\t" + str(prediction) |