Downloads · 30 days
3
16% of all-time downloads
seleznevchamp1/rubert-tiny2-NER-augmented
rubert-tiny2-NER-augmented is a machine learning model from seleznevchamp1. Use it for the machine learning task on the model card, and read the license before you ship it in a product. The card lists the license as mit.
--- language: ru tags: - NER - BERT - augmentation - ruBERT license: mit datasets: - custom: hh.ru metrics: - f1 ---
Downloads · 30 days
3
16% of all-time downloads
All-time downloads
19
Public
Parameters
29.1M
116 MB on disk
Likes
0
Public
Click a slice to open those files.
.safetensors116 MB · 97%
From the Hugging Face model README
language: ru tags:
Compact ruBERT model fine-tuned with data augmentation to extract Skill and Knowledge entities.
Base model: cointegrated/ruBert-tiny2
Training: 250 manually annotated + 250 augmented samples (total 500).
F1: 0.52
from transformers import AutoTokenizer, AutoModelForTokenClassification, pipeline
model = AutoModelForTokenClassification.from_pretrained("seleznevchamp1/rubert-tiny2-NER-augmented")
tok = AutoTokenizer.from_pretrained("seleznevchamp1/rubert-tiny2-NER-augmented")
nlp = pipeline("ner", model=model, tokenizer=tok)
print(nlp("Опыт с Python и управление командой"))