From aa2eac49c34822e1910e28afa39f3b069c9f4aa1 Mon Sep 17 00:00:00 2001 From: kubapok Date: Thu, 10 Feb 2022 14:37:36 +0100 Subject: [PATCH] a --- .../config.py | 18 ++++++++++++++++++ .../run.sh | 2 +- 2 files changed, 19 insertions(+), 1 deletion(-) create mode 100644 roberta_year_as_token_better_finetunning/config.py diff --git a/roberta_year_as_token_better_finetunning/config.py b/roberta_year_as_token_better_finetunning/config.py new file mode 100644 index 0000000..4b62f72 --- /dev/null +++ b/roberta_year_as_token_better_finetunning/config.py @@ -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' diff --git a/roberta_year_as_token_better_finetunning/run.sh b/roberta_year_as_token_better_finetunning/run.sh index c07fa93..3d6e885 100644 --- a/roberta_year_as_token_better_finetunning/run.sh +++ b/roberta_year_as_token_better_finetunning/run.sh @@ -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