metadata

library_name: transformers
license: apache-2.0
language:
  - ru
metrics:
  - f1
  - accuracy
base_model: ai-forever/ruRoberta-large

Model Card for Model ID

Model Details

Model Description

Language(s) (NLP): Russian
License: apache-2.0
Finetuned from model: ai-forever/ruRoberta-large

Usage

from transformers import AutoTokenizer, AutoModelForSequenceClassification

model = AutoModelForSequenceClassification.from_pretrained("pasukka/auto-parts-term-classifier-v.3")
tokenizer = AutoTokenizer.from_pretrained('ai-forever/ruRoberta-large')

term = 'фильтр топливный'
outputs = model.forward(**tokenizer(term, return_tensors='pt').to(device='cuda'))
id = outputs.logits.argmax(dim=1)[0].item()

print(model.config.id2label[id])

Result:

фильтр

pasukka
/

auto-parts-term-classifier-v.3

Model Card for Model ID

Model Details

Model Description

Usage

Metrics