From 2898bee8e9fc29aa1889af5ba2a3c8977c97040e Mon Sep 17 00:00:00 2001 From: Jakub Pokrywka Date: Thu, 4 Nov 2021 12:09:55 +0100 Subject: [PATCH] a --- roberta_ft/04_predict.py | 2 +- roberta_ft/run.sh | 2 +- roberta_with_year_ft/04_predict.py | 2 +- 3 files changed, 3 insertions(+), 3 deletions(-) diff --git a/roberta_ft/04_predict.py b/roberta_ft/04_predict.py index 2e45f57..1eb9956 100644 --- a/roberta_ft/04_predict.py +++ b/roberta_ft/04_predict.py @@ -19,7 +19,7 @@ def write(f_path_in, f_path_out): a = get_formatted(l_in) f_out.write(a + '\n') -model = 'robertamodel/checkpoint-24000' +model = 'robertamodel' unmasker = pipeline('fill-mask', model=model) write('dev-0_in.csv', '../dev-0/out.tsv') write('test-A_in.csv', '../test-A/out.tsv') diff --git a/roberta_ft/run.sh b/roberta_ft/run.sh index ec35538..a35e620 100644 --- a/roberta_ft/run.sh +++ b/roberta_ft/run.sh @@ -1,3 +1,3 @@ python 00_create_datasets.py -python 04_predict.py bash 2_run.sh +python 04_predict.py diff --git a/roberta_with_year_ft/04_predict.py b/roberta_with_year_ft/04_predict.py index 622616f..1eb9956 100644 --- a/roberta_with_year_ft/04_predict.py +++ b/roberta_with_year_ft/04_predict.py @@ -19,7 +19,7 @@ def write(f_path_in, f_path_out): a = get_formatted(l_in) f_out.write(a + '\n') -model = 'robertamodel/checkpoint-60000' +model = 'robertamodel' unmasker = pipeline('fill-mask', model=model) write('dev-0_in.csv', '../dev-0/out.tsv') write('test-A_in.csv', '../test-A/out.tsv')