Views
No views yet
1文脈:{context}
2読み:{reading} [SEP] 候補:{candidate}trust_remote_code=True.1from transformers import AutoModelForSequenceClassification, AutoTokenizer
2
3model_id = "fa0311/ime-context-reranker-v1"
4tokenizer = AutoTokenizer.from_pretrained(model_id, trust_remote_code=True)
5model = AutoModelForSequenceClassification.from_pretrained(model_id)
6
7query = "文脈:家に\n読み:かえる"
8candidate = "候補:帰る"
9inputs = tokenizer(query, candidate, return_tensors="pt")
10score = model(**inputs).logits.item()| Evaluation set | Top-1 | Top-5 |
|---|---|---|
| Development | 94.83% | 100.00% |
| Locked 276 | 80.80% | 99.28% |
| Curated 1,416 | 82.20% | 99.44% |
| Difficult full-input 805 | 42.11% | 62.48% |
line-corporation/line-distilbert-base-japanese,
which is also distributed under Apache-2.0.