19 lines
316 B
Python
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'
|