ireland-news-headlines/roberta_year_as_text_better_finetunning_10k/config.py
2022-02-20 22:03:28 +01:00

19 lines
316 B
Python

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'