ireland-news-headlines/roberta_no_year/config.py

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'