Views
No views yet
1from ruscxnpipe import ConstructionClassifier
2
3classifier = ConstructionClassifier(
4 model_name="Futyn-Maker/ruscxn-classifier"
5)
6
7# Classify candidates (output from semantic search)
8queries = ["Петр так и замер."]
9candidates = [[{"id": "pattern1", "pattern": "NP-Nom так и VP-Pfv"}]]
10
11results = classifier.classify_candidates(queries, candidates)
12print(results[0][0]['is_present']) # 1 if present, 0 if absent1from transformers import AutoTokenizer, AutoModelForSequenceClassification
2import torch
3
4model = AutoModelForSequenceClassification.from_pretrained("Futyn-Maker/ruscxn-classifier")
5tokenizer = AutoTokenizer.from_pretrained("Futyn-Maker/ruscxn-classifier")
6
7# Format: "passage: [pattern][Sep]query: [example]"
8text = "passage: NP-Nom так и VP-Pfv[Sep]query: Петр так и замер."
9inputs = tokenizer(text, return_tensors="pt", truncation=True)
10
11with torch.no_grad():
12 outputs = model(**inputs)
13 prediction = torch.softmax(outputs.logits, dim=-1)
14 is_present = torch.argmax(prediction, dim=-1).item()
15
16print(f"Construction present: {is_present}") # 1 = present, 0 = absent"passage: [pattern][Sep]query: [example]"