Downloads · 30 days
4
14% of all-time downloads
seleznevchamp1/rubert-tiny2-NER-no_aug
rubert-tiny2-NER-no_aug 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 - ruBERT - job-postings license: mit datasets: - custom: hh.ru metrics: - f1 ---
Downloads · 30 days
4
14% of all-time downloads
All-time downloads
29
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 transformer model fine-tuned for extracting Skill and Knowledge entities from Russian job descriptions.
Base model: cointegrated/ruBert-tiny2
Training: fine-tuned on 250 manually annotated vacancies (no augmentation).
Labels: O, B-Skill, I-Skill, B-Knowledge, I-Knowledge
F1: 0.38
from transformers import AutoTokenizer, AutoModelForTokenClassification, pipeline
model = AutoModelForTokenClassification.from_pretrained("seleznevchamp1/rubert-tiny2-NER-no_aug")
tok = AutoTokenizer.from_pretrained("seleznevchamp1/rubert-tiny2-NER-no_aug")
nlp = pipeline("ner", model=model, tokenizer=tok)
print(nlp("Требуется знание Python и умение управлять командой"))