Views
No views yet
uzbtrust/uzbek-operator-ner
on GitHub.| Test set | F1 |
|---|---|
| CoNLL-2003 (English) | 0.786 |
| WikiANN (Russian) | 0.817 |
| Operator domain (synthetic) | 1.000 |
model.pt — {"epoch", "best_f1", "model": state_dict}word_vocab.json, char_vocab.json, tag_map.json — vocabularies built during trainingconfig.json — architecture summarystate_dict, not a transformers-compatible checkpoint):1git clone https://github.com/uzbtrust/uzbek-operator-ner
2cd uzbek-operator-ner1import torch, json
2from model.ner_model import NERModel # see repo for exact constructor args
3
4vocab = json.load(open("word_vocab.json"))
5tags = json.load(open("tag_map.json"))
6
7ckpt = torch.load("model.pt", map_location="cpu")
8model = NERModel(vocab_size=len(vocab), num_tags=len(tags))
9model.load_state_dict(ckpt["model"])
10model.eval()training/predict.py
in the GitHub repo for a complete, runnable inference example.