ireland-news-headlines/roberta_year_as_token_better_finetunning/config.py

19 lines
316 B
Python
Raw Normal View History

2022-02-10 14:37:36 +01:00
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'