kubapok/ireland-news-word-gap-prediction
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

a

Browse Source
This commit is contained in:
Jakub Pokrywka 2021-11-04 12:09:55 +01:00
parent 5308358ad7
commit 2898bee8e9
3 changed files with 3 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
roberta_ft/04_predict.py
View File

@ -19,7 +19,7 @@ def write(f_path_in, f_path_out):
a = get_formatted(l_in) a = get_formatted(l_in)
f_out.write(a + '\n') f_out.write(a + '\n')
model = 'robertamodel/checkpoint-24000' model = 'robertamodel'
unmasker = pipeline('fill-mask', model=model) unmasker = pipeline('fill-mask', model=model)
write('dev-0_in.csv', '../dev-0/out.tsv') write('dev-0_in.csv', '../dev-0/out.tsv')
write('test-A_in.csv', '../test-A/out.tsv') write('test-A_in.csv', '../test-A/out.tsv')

2
roberta_ft/run.sh
View File

@ -1,3 +1,3 @@
python 00_create_datasets.py python 00_create_datasets.py
python 04_predict.py
bash 2_run.sh bash 2_run.sh
python 04_predict.py

2
roberta_with_year_ft/04_predict.py
View File

@ -19,7 +19,7 @@ def write(f_path_in, f_path_out):
a = get_formatted(l_in) a = get_formatted(l_in)
f_out.write(a + '\n') f_out.write(a + '\n')
model = 'robertamodel/checkpoint-60000' model = 'robertamodel'
unmasker = pipeline('fill-mask', model=model) unmasker = pipeline('fill-mask', model=model)
write('dev-0_in.csv', '../dev-0/out.tsv') write('dev-0_in.csv', '../dev-0/out.tsv')
write('test-A_in.csv', '../test-A/out.tsv') write('test-A_in.csv', '../test-A/out.tsv')