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

a

Browse Source
This commit is contained in:
kubapok 2022-02-10 14:37:36 +01:00
parent 92b4eafb30
commit aa2eac49c3
2 changed files with 19 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

18
roberta_year_as_token_better_finetunning/config.py Normal file
View File

@ -0,0 +1,18 @@
LABELS_DICT = {'news':0,
'sport':1,
'business':2,
'opinion':3,
'culture':4,
'lifestyle':5,
'removed':6}
LABELS_LIST = ['news',
'sport',
'business',
'opinion',
'culture',
'lifestyle',
'removed']
MODEL = 'roberta-base'

2
roberta_year_as_token_better_finetunning/run.sh
View File

@ -1 +1 @@
python run_glue.py --model_name_or_path roberta-base --train_file /media/kuba/ssdsam/gonito/ireland-news/train/huggingface_format_year.csv --validation_file /media/kuba/ssdsam/gonito/ireland-news/dev-0/huggingface_format_year.csv --do_train --do_eval --max_seq_length 64 --per_device_train_batch_size 32 --learning_rate 2e-5 --num_train_epochs 3 --save_steps=5000 --eval_steps=5000 --evaluation_strategy steps --output_dir ./roberta-ireland python run_glue.py --model_name_or_path roberta-base --train_file ../train/huggingface_format_year.csv --validation_file ../dev-0/huggingface_format_year.csv --do_train --do_eval --max_seq_length 64 --per_device_train_batch_size 32 --learning_rate 2e-5 --num_train_epochs 3 --save_steps=5000 --eval_steps=5000 --evaluation_strategy steps --output_dir ./roberta-ireland