AB
Size: a a a
AB
SB
A
DD
DD
from transformers import AutoModel, AutoModelForSequenceClassification
model = AutoModel.from_pretrained('cointegrated/rubert-tiny-toxicity')
model.save_pretrained('tmp')
model = AutoModelForSequenceClassification.from_pretrained('tmp', num_labels=3)
DD
tokenized_datasets = raw_datasets.map(tokenize_function, batched=True).map(lambda x: {'labels': [y-1 for y in x['labels']]}, batched=True)tokenized_datasets = raw_datasets.map(tokenize_function, batched=True)
A
DD
A
DK
AT
AV
TM
M
ni
SS
ni
M
M