cstr commited on
Commit
a3c98e7
·
verified ·
1 Parent(s): d20fdac

Add model card README

Browse files
Files changed (1) hide show
  1. README.md +59 -0
README.md ADDED
@@ -0,0 +1,59 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ license: mit
3
+ tags:
4
+ - gguf
5
+ - bert
6
+ - ner
7
+ - token-classification
8
+ - named-entity-recognition
9
+ base_model: dslim/bert-base-NER
10
+ pipeline_tag: token-classification
11
+ ---
12
+
13
+ # BERT Base NER — GGUF
14
+
15
+ GGUF conversion of [dslim/bert-base-NER](https://huggingface.co/dslim/bert-base-NER) for use with [CrispEmbed](https://github.com/CrispStrobe/CrispEmbed).
16
+
17
+ Fixed-label Named Entity Recognition on English text. BERT-base-cased (110M params) fine-tuned on CoNLL-03 with 9 IOB labels.
18
+
19
+ ## Labels
20
+
21
+ | ID | Label | Description |
22
+ |----|-------|-------------|
23
+ | 0 | O | Outside any entity |
24
+ | 1 | B-MISC | Beginning of miscellaneous entity |
25
+ | 2 | I-MISC | Inside miscellaneous entity |
26
+ | 3 | B-PER | Beginning of person name |
27
+ | 4 | I-PER | Inside person name |
28
+ | 5 | B-ORG | Beginning of organization |
29
+ | 6 | I-ORG | Inside organization |
30
+ | 7 | B-LOC | Beginning of location |
31
+ | 8 | I-LOC | Inside location |
32
+
33
+ ## Available Formats
34
+
35
+ | File | Format | Size |
36
+ |------|--------|------|
37
+ | `bert-base-ner-f32.gguf` | Float32 | 412 MB |
38
+ | `bert-base-ner-q8_0.gguf` | Q8_0 | 111 MB |
39
+ | `bert-base-ner-q4_k.gguf` | Q4_K | 70 MB |
40
+
41
+ ## Usage
42
+
43
+ ```bash
44
+ crispembed -m bert-base-ner-q8_0.gguf --ner "Barack Obama was born in Hawaii"
45
+ ```
46
+
47
+ ```python
48
+ from crispembed import CrispNER
49
+ ner = CrispNER("bert-base-ner-q8_0.gguf")
50
+ entities = ner.extract("Barack Obama was born in Hawaii")
51
+ # [{"text": "Barack Obama", "label": "PER", "start": 0, "end": 12, "score": 0.999},
52
+ # {"text": "Hawaii", "label": "LOC", "start": 25, "end": 31, "score": 1.000}]
53
+ ```
54
+
55
+ Auto-detected as BERT NER (vs GLiNER zero-shot) from `ner.classifier.weight` in GGUF.
56
+
57
+ ## Parity
58
+
59
+ Encoder output: cos_min=0.999971 vs HuggingFace transformers (F32).