Downloads · 30 days
14
10% of all-time downloads
Futyn-Maker/ruscxn-classifier
ruscxn-classifier is a text classification model from Futyn-Maker. Use it when you need a label for a piece of text. It is set up for transformers.
A binary classification model for determining whether a Russian Constructicon pattern is present in a given text example. Fine-tuned from intfloat/multilingual-e5-large in two stages: first as a semantic model on Russ…
Downloads · 30 days
14
10% of all-time downloads
All-time downloads
147
Public
Parameters
560M
2.3 GB on disk
Likes
0
Public
Click a slice to open those files.
.safetensors2.2 GB · 99%
From the Hugging Face model README
A binary classification model for determining whether a Russian Constructicon pattern is present in a given text example. Fine-tuned from intfloat/multilingual-e5-large in two stages: first as a semantic model on Russian Constructicon data, then for binary classification.
This model is designed for use with the RusCxnPipe library:
from ruscxnpipe import ConstructionClassifier
classifier = ConstructionClassifier(
model_name="Futyn-Maker/ruscxn-classifier"
)
# Classify candidates (output from semantic search)
queries = ["Петр так и замер."]
candidates = [[{"id": "pattern1", "pattern": "NP-Nom так и VP-Pfv"}]]
results = classifier.classify_candidates(queries, candidates)
print(results[0][0]['is_present']) # 1 if present, 0 if absent
from transformers import AutoTokenizer, AutoModelForSequenceClassification
import torch
model = AutoModelForSequenceClassification.from_pretrained("Futyn-Maker/ruscxn-classifier")
tokenizer = AutoTokenizer.from_pretrained("Futyn-Maker/ruscxn-classifier")
# Format: "passage: [pattern][Sep]query: [example]"
text = "passage: NP-Nom так и VP-Pfv[Sep]query: Петр так и замер."
inputs = tokenizer(text, return_tensors="pt", truncation=True)
with torch.no_grad():
outputs = model(**inputs)
prediction = torch.softmax(outputs.logits, dim=-1)
is_present = torch.argmax(prediction, dim=-1).item()
print(f"Construction present: {is_present}") # 1 = present, 0 = absent
The model expects input in the format: "passage: [pattern][Sep]query: [example]"