Finalize TypePredictor model card and release artifacts
Browse files- README.md +320 -66
- release_manifest.json +119 -119
README.md
CHANGED
|
@@ -9,6 +9,7 @@ tags:
|
|
| 9 |
- entity-typing
|
| 10 |
- wojood
|
| 11 |
- neoarabert
|
|
|
|
| 12 |
datasets:
|
| 13 |
- U4RASD/TypePrediction
|
| 14 |
metrics:
|
|
@@ -18,31 +19,59 @@ metrics:
|
|
| 18 |
|
| 19 |
# TypePredictor
|
| 20 |
|
| 21 |
-
`TypePredictor` is a mention-level Arabic entity
|
| 22 |
-
|
| 23 |
-
|
| 24 |
-
of 21 coarse Wojood entity types.
|
| 25 |
|
| 26 |
-
|
| 27 |
-
|
| 28 |
-
|
| 29 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 30 |
|
| 31 |
## Architecture
|
| 32 |
|
| 33 |
```text
|
| 34 |
-
sentence + known span
|
| 35 |
-
-> insert [ENT]
|
| 36 |
-
-> U4RASD/NeoAraBERT
|
| 37 |
-
->
|
| 38 |
-
->
|
|
|
|
| 39 |
-> Linear(768, 21)
|
| 40 |
-> argmax entity type
|
| 41 |
```
|
| 42 |
|
| 43 |
-
|
| 44 |
-
|
| 45 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 46 |
|
| 47 |
## Labels
|
| 48 |
|
|
@@ -50,64 +79,248 @@ GPE, ORG, DATE, PERS, NORP, ORDINAL, OCC, EVENT, CARDINAL, LOC, WEBSITE, FAC, LA
|
|
| 50 |
|
| 51 |
## Dataset
|
| 52 |
|
| 53 |
-
-
|
| 54 |
-
-
|
| 55 |
-
-
|
| 56 |
-
-
|
| 57 |
-
-
|
| 58 |
-
-
|
| 59 |
-
- Test
|
|
|
|
|
|
|
| 60 |
|
| 61 |
-
The split is mention-level
|
| 62 |
-
|
| 63 |
-
therefore be interpreted separately.
|
| 64 |
|
| 65 |
-
|
|
|
|
|
|
|
| 66 |
|
| 67 |
-
|
| 68 |
-
|---|---:|---:|---:|---:|---:|
|
| 69 |
-
| overall | 12600 | 0.977857 | 0.977857 | 0.958728 | 0.977791 |
|
| 70 |
-
| unseen_sentence | 939 | 0.945687 | 0.945687 | 0.908841 | 0.945011 |
|
| 71 |
-
| seen_sentence_new_entity | 11661 | 0.980448 | 0.980448 | 0.963262 | 0.980403 |
|
| 72 |
|
| 73 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 74 |
|
| 75 |
-
|
| 76 |
-
|---|---:|---:|---:|---:|---:|
|
| 77 |
-
| overall | 12600 | 0.979365 | 0.979365 | 0.959920 | 0.979297 |
|
| 78 |
-
| unseen_sentence | 994 | 0.962777 | 0.962777 | 0.893174 | 0.961780 |
|
| 79 |
-
| seen_sentence_new_entity | 11606 | 0.980786 | 0.980786 | 0.964113 | 0.980712 |
|
| 80 |
|
| 81 |
-
|
| 82 |
-
|
|
|
|
|
|
|
|
|
|
| 83 |
|
| 84 |
## Checkpoints
|
| 85 |
|
| 86 |
-
- Best checkpoint:
|
|
|
|
|
|
|
|
|
|
| 87 |
- Best validation macro F1: `0.958728`
|
| 88 |
-
- Latest checkpoint:
|
| 89 |
-
-
|
| 90 |
-
-
|
|
|
|
|
|
|
|
|
|
| 91 |
|
| 92 |
-
|
| 93 |
-
output directory.
|
| 94 |
|
| 95 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 96 |
|
| 97 |
-
|
| 98 |
-
|
| 99 |
-
|
| 100 |
-
|
| 101 |
-
|
| 102 |
-
|
| 103 |
-
|
| 104 |
-
|
| 105 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 106 |
|
| 107 |
## Loading
|
| 108 |
|
| 109 |
-
|
| 110 |
-
|
| 111 |
|
| 112 |
```python
|
| 113 |
from huggingface_hub import hf_hub_download
|
|
@@ -117,20 +330,61 @@ source = hf_hub_download("U4RASD/TypePredictor", "modeling_type_predictor.py")
|
|
| 117 |
spec = importlib.util.spec_from_file_location("modeling_type_predictor", source)
|
| 118 |
module = importlib.util.module_from_spec(spec)
|
| 119 |
spec.loader.exec_module(module)
|
|
|
|
| 120 |
model, tokenizer, config = module.NeoAraBERTTypePredictor.from_pretrained(
|
| 121 |
"U4RASD/TypePredictor"
|
| 122 |
)
|
|
|
|
| 123 |
```
|
| 124 |
|
| 125 |
-
##
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 126 |
|
| 127 |
-
|
| 128 |
-
|
|
|
|
| 129 |
|
| 130 |
## Limitations
|
| 131 |
|
| 132 |
-
-
|
| 133 |
-
-
|
| 134 |
-
|
| 135 |
-
-
|
| 136 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 9 |
- entity-typing
|
| 10 |
- wojood
|
| 11 |
- neoarabert
|
| 12 |
+
- mention-classification
|
| 13 |
datasets:
|
| 14 |
- U4RASD/TypePrediction
|
| 15 |
metrics:
|
|
|
|
| 19 |
|
| 20 |
# TypePredictor
|
| 21 |
|
| 22 |
+
`TypePredictor` is a mention-level Arabic entity type classifier. It assumes the
|
| 23 |
+
entity span is already known, inserts `[ENT]` and `[/ENT]` around that span, and
|
| 24 |
+
predicts one of 21 Wojood-style entity types.
|
|
|
|
| 25 |
|
| 26 |
+
This model is intended as a type-normalization component for a later
|
| 27 |
+
relation-extraction pipeline, where relation subjects and objects are already
|
| 28 |
+
available as spans and need a consistent coarse entity type.
|
| 29 |
+
|
| 30 |
+
## Key result
|
| 31 |
+
|
| 32 |
+
The repository root contains the best checkpoint selected by overall validation
|
| 33 |
+
macro F1. The best checkpoint was step `25,000` / epoch
|
| 34 |
+
`3.968292`.
|
| 35 |
+
|
| 36 |
+
| Split | Rows | Accuracy | Micro F1 | Macro F1 present types | Macro F1 all 21 | Weighted F1 |
|
| 37 |
+
|---|---:|---:|---:|---:|---:|---:|
|
| 38 |
+
| Validation overall | 12,600 | 0.977857 | 0.977857 | 0.958728 | 0.958728 | 0.977791 |
|
| 39 |
+
| Test overall | 12,600 | 0.979365 | 0.979365 | 0.959920 | 0.959920 | 0.979297 |
|
| 40 |
+
|
| 41 |
+
## What this model does and does not do
|
| 42 |
+
|
| 43 |
+
- It classifies a supplied mention/span.
|
| 44 |
+
- It does not detect entity boundaries.
|
| 45 |
+
- It does not include a `NONE` class because the training, validation, and test
|
| 46 |
+
splits do not contain `NONE` examples.
|
| 47 |
+
- It uses one shared classifier for all mentions; there is no separate subject
|
| 48 |
+
or object head.
|
| 49 |
|
| 50 |
## Architecture
|
| 51 |
|
| 52 |
```text
|
| 53 |
+
Arabic sentence + known character span
|
| 54 |
+
-> insert [ENT] and [/ENT] around the exact span
|
| 55 |
+
-> tokenize with U4RASD/NeoAraBERT tokenizer
|
| 56 |
+
-> U4RASD/NeoAraBERT encoder
|
| 57 |
+
-> final hidden state at CLS position
|
| 58 |
+
-> Dropout(0.10)
|
| 59 |
-> Linear(768, 21)
|
| 60 |
-> argmax entity type
|
| 61 |
```
|
| 62 |
|
| 63 |
+
Architecture details:
|
| 64 |
+
|
| 65 |
+
- Base encoder: `U4RASD/NeoAraBERT`
|
| 66 |
+
- Hidden size: `768`
|
| 67 |
+
- Tokenizer size after markers: `65,002`
|
| 68 |
+
- Classifier shape: `768 -> 21`
|
| 69 |
+
- Total parameters: `248,162,325`
|
| 70 |
+
- Encoder parameters: `248,146,176`
|
| 71 |
+
- Classifier parameters: `16,149`
|
| 72 |
+
- Loss: ordinary unweighted multiclass cross-entropy
|
| 73 |
+
- No class weights, focal loss, oversampling, weighted sampler, span pooling,
|
| 74 |
+
threshold, extra MLP, or two-head design.
|
| 75 |
|
| 76 |
## Labels
|
| 77 |
|
|
|
|
| 79 |
|
| 80 |
## Dataset
|
| 81 |
|
| 82 |
+
- Dataset repository: `U4RASD/TypePrediction`
|
| 83 |
+
- Train file: `type_predictor_train.jsonl`
|
| 84 |
+
- Validation file: `type_predictor_val.jsonl`
|
| 85 |
+
- Test file: `type_predictor_test.jsonl`
|
| 86 |
+
- Train rows: `100,796`
|
| 87 |
+
- Validation rows: `12,600`
|
| 88 |
+
- Test rows: `12,600`
|
| 89 |
+
- Dataset validation timestamp: `2026-07-12T15:30:12+00:00`
|
| 90 |
+
- IDs unique across splits: `True`
|
| 91 |
|
| 92 |
+
The split is mention-level. A sentence can appear in more than one split with a
|
| 93 |
+
different target mention. For that reason, the model card reports both:
|
|
|
|
| 94 |
|
| 95 |
+
- `seen_sentence_new_entity`: the sentence text was seen in training, but the
|
| 96 |
+
evaluated target mention is new.
|
| 97 |
+
- `unseen_sentence`: the sentence text was not seen in training.
|
| 98 |
|
| 99 |
+
## Preprocessing and encoding
|
|
|
|
|
|
|
|
|
|
|
|
|
| 100 |
|
| 101 |
+
- Markers: `[ENT]` and `[/ENT]`
|
| 102 |
+
- Marker insertion is based on exact character offsets, not string replacement.
|
| 103 |
+
- Maximum sequence length: `512`
|
| 104 |
+
- Default context window: `300` characters
|
| 105 |
+
- Fallback context candidates: `None, 500, 300, 150, 80, 30, 0`
|
| 106 |
+
- The encoder rejects examples where truncation fails to preserve exactly one
|
| 107 |
+
opening marker and one closing marker in the correct order.
|
| 108 |
+
|
| 109 |
+
## Training configuration
|
| 110 |
+
|
| 111 |
+
| Setting | Value |
|
| 112 |
+
|---|---:|
|
| 113 |
+
| Seed | `42` |
|
| 114 |
+
| Epochs | `4.0` |
|
| 115 |
+
| Train batch size/device | `4` |
|
| 116 |
+
| Eval batch size/device | `8` |
|
| 117 |
+
| Gradient accumulation | `4` |
|
| 118 |
+
| Effective batch size | `16` |
|
| 119 |
+
| Encoder learning rate | `1e-05` |
|
| 120 |
+
| Classifier learning rate | `5e-05` |
|
| 121 |
+
| Weight decay | `0.01` |
|
| 122 |
+
| Warmup ratio | `0.1` |
|
| 123 |
+
| Max grad norm | `1.0` |
|
| 124 |
+
| Dropout | `0.1` |
|
| 125 |
+
| Logging steps | `50` |
|
| 126 |
+
| Eval steps | `500` |
|
| 127 |
+
| Save steps | `500` |
|
| 128 |
+
| FP16 | `True` |
|
| 129 |
+
| Best model criterion | overall validation macro F1 |
|
| 130 |
+
|
| 131 |
+
Training runtime:
|
| 132 |
+
|
| 133 |
+
- Started: `2026-07-12T15:30:38+00:00`
|
| 134 |
+
- Finished: `2026-07-12T18:55:01+00:00`
|
| 135 |
+
- Wall time: `12262.47` seconds (`3.41` hours)
|
| 136 |
+
- Train runtime reported by Trainer: `12261.72` seconds
|
| 137 |
+
- Train samples/sec: `32.882`
|
| 138 |
+
- Train steps/sec: `2.055`
|
| 139 |
+
- Final train loss: `0.190693`
|
| 140 |
|
| 141 |
+
Hardware/runtime:
|
|
|
|
|
|
|
|
|
|
|
|
|
| 142 |
|
| 143 |
+
- GPU: `NVIDIA A40`
|
| 144 |
+
- GPU VRAM: `44.43 GiB`
|
| 145 |
+
- PyTorch: `2.5.1+cu124`
|
| 146 |
+
- Transformers: `4.49.0`
|
| 147 |
+
- Python: `3.12.3`
|
| 148 |
|
| 149 |
## Checkpoints
|
| 150 |
|
| 151 |
+
- Best checkpoint source: `outputs/TypePredictor/checkpoints/checkpoint-25000`
|
| 152 |
+
- Preserved best checkpoint: `outputs/TypePredictor/best_checkpoint`
|
| 153 |
+
- Best step: `25,000`
|
| 154 |
+
- Best epoch: `3.968292`
|
| 155 |
- Best validation macro F1: `0.958728`
|
| 156 |
+
- Latest completed checkpoint source: `outputs/TypePredictor/checkpoints/checkpoint-25196`
|
| 157 |
+
- Preserved latest checkpoint: `outputs/TypePredictor/latest_checkpoint`
|
| 158 |
+
- Latest step: `25,196`
|
| 159 |
+
- Latest epoch: `3.999405`
|
| 160 |
+
- Repository root released model: `best checkpoint`
|
| 161 |
+
- `checkpoints/latest/` contains the latest completed checkpoint snapshot.
|
| 162 |
|
| 163 |
+
## Validation results
|
|
|
|
| 164 |
|
| 165 |
+
| Subset | Rows | Accuracy | Micro F1 | Macro F1 present types | Macro F1 all 21 | Weighted F1 |
|
| 166 |
+
|---|---:|---:|---:|---:|---:|---:|
|
| 167 |
+
| overall | 12,600 | 0.977857 | 0.977857 | 0.958728 | 0.958728 | 0.977791 |
|
| 168 |
+
| unseen_sentence | 939 | 0.945687 | 0.945687 | 0.908841 | 0.865563 | 0.945011 |
|
| 169 |
+
| seen_sentence_new_entity | 11,661 | 0.980448 | 0.980448 | 0.963262 | 0.963262 | 0.980403 |
|
| 170 |
+
|
| 171 |
+
## Test results
|
| 172 |
+
|
| 173 |
+
| Subset | Rows | Accuracy | Micro F1 | Macro F1 present types | Macro F1 all 21 | Weighted F1 |
|
| 174 |
+
|---|---:|---:|---:|---:|---:|---:|
|
| 175 |
+
| overall | 12,600 | 0.979365 | 0.979365 | 0.959920 | 0.959920 | 0.979297 |
|
| 176 |
+
| unseen_sentence | 994 | 0.962777 | 0.962777 | 0.893174 | 0.850642 | 0.961780 |
|
| 177 |
+
| seen_sentence_new_entity | 11,606 | 0.980786 | 0.980786 | 0.964113 | 0.964113 | 0.980712 |
|
| 178 |
+
|
| 179 |
+
## Per-class validation results
|
| 180 |
+
|
| 181 |
+
### Validation overall
|
| 182 |
+
|
| 183 |
+
| Label | Precision | Recall | F1 | Support |
|
| 184 |
+
|---|---:|---:|---:|---:|
|
| 185 |
+
| GPE | 0.977432 | 0.982156 | 0.979789 | 2,690 |
|
| 186 |
+
| ORG | 0.972832 | 0.982707 | 0.977745 | 2,660 |
|
| 187 |
+
| DATE | 0.993431 | 0.994939 | 0.994185 | 1,976 |
|
| 188 |
+
| PERS | 0.973832 | 0.972015 | 0.972923 | 1,072 |
|
| 189 |
+
| NORP | 0.975050 | 0.955969 | 0.965415 | 1,022 |
|
| 190 |
+
| ORDINAL | 0.979513 | 0.980769 | 0.980141 | 780 |
|
| 191 |
+
| OCC | 0.981889 | 0.986996 | 0.984436 | 769 |
|
| 192 |
+
| EVENT | 0.967302 | 0.941645 | 0.954301 | 377 |
|
| 193 |
+
| CARDINAL | 0.973333 | 0.978552 | 0.975936 | 373 |
|
| 194 |
+
| LOC | 0.974576 | 0.962343 | 0.968421 | 239 |
|
| 195 |
+
| WEBSITE | 0.986577 | 0.993243 | 0.989899 | 148 |
|
| 196 |
+
| FAC | 0.916667 | 0.930769 | 0.923664 | 130 |
|
| 197 |
+
| LAW | 1.000000 | 1.000000 | 1.000000 | 90 |
|
| 198 |
+
| TIME | 0.974359 | 0.873563 | 0.921212 | 87 |
|
| 199 |
+
| MONEY | 0.933333 | 1.000000 | 0.965517 | 42 |
|
| 200 |
+
| CURR | 1.000000 | 0.951220 | 0.975000 | 41 |
|
| 201 |
+
| LANGUAGE | 0.933333 | 0.848485 | 0.888889 | 33 |
|
| 202 |
+
| PERCENT | 1.000000 | 0.903226 | 0.949153 | 31 |
|
| 203 |
+
| PRODUCT | 1.000000 | 0.842105 | 0.914286 | 19 |
|
| 204 |
+
| QUANTITY | 1.000000 | 0.818182 | 0.900000 | 11 |
|
| 205 |
+
| UNIT | 0.909091 | 1.000000 | 0.952381 | 10 |
|
| 206 |
+
|
| 207 |
+
## Per-class test results
|
| 208 |
+
|
| 209 |
+
### Test overall
|
| 210 |
+
|
| 211 |
+
| Label | Precision | Recall | F1 | Support |
|
| 212 |
+
|---|---:|---:|---:|---:|
|
| 213 |
+
| GPE | 0.973897 | 0.984392 | 0.979117 | 2,691 |
|
| 214 |
+
| ORG | 0.980769 | 0.978187 | 0.979477 | 2,659 |
|
| 215 |
+
| DATE | 0.990438 | 0.995448 | 0.992936 | 1,977 |
|
| 216 |
+
| PERS | 0.979535 | 0.983193 | 0.981361 | 1,071 |
|
| 217 |
+
| NORP | 0.976540 | 0.978452 | 0.977495 | 1,021 |
|
| 218 |
+
| ORDINAL | 0.982097 | 0.983355 | 0.982726 | 781 |
|
| 219 |
+
| OCC | 0.987047 | 0.989610 | 0.988327 | 770 |
|
| 220 |
+
| EVENT | 0.986413 | 0.962865 | 0.974497 | 377 |
|
| 221 |
+
| CARDINAL | 0.977901 | 0.951613 | 0.964578 | 372 |
|
| 222 |
+
| LOC | 0.969565 | 0.933054 | 0.950959 | 239 |
|
| 223 |
+
| WEBSITE | 0.966887 | 0.986486 | 0.976589 | 148 |
|
| 224 |
+
| FAC | 0.928000 | 0.899225 | 0.913386 | 129 |
|
| 225 |
+
| LAW | 1.000000 | 1.000000 | 1.000000 | 91 |
|
| 226 |
+
| TIME | 0.915663 | 0.873563 | 0.894118 | 87 |
|
| 227 |
+
| MONEY | 0.973684 | 0.880952 | 0.925000 | 42 |
|
| 228 |
+
| CURR | 0.975610 | 0.975610 | 0.975610 | 41 |
|
| 229 |
+
| LANGUAGE | 0.875000 | 0.848485 | 0.861538 | 33 |
|
| 230 |
+
| PERCENT | 0.964286 | 0.870968 | 0.915254 | 31 |
|
| 231 |
+
| PRODUCT | 1.000000 | 0.947368 | 0.972973 | 19 |
|
| 232 |
+
| QUANTITY | 0.909091 | 1.000000 | 0.952381 | 10 |
|
| 233 |
+
| UNIT | 1.000000 | 1.000000 | 1.000000 | 11 |
|
| 234 |
|
| 235 |
+
### Test unseen-sentence per-class results
|
| 236 |
+
|
| 237 |
+
This subset is the strictest split because the full sentence is unseen during
|
| 238 |
+
training. Some rare labels have very small support here, so their F1 scores are
|
| 239 |
+
high variance.
|
| 240 |
+
|
| 241 |
+
### Test unseen_sentence
|
| 242 |
+
|
| 243 |
+
| Label | Precision | Recall | F1 | Support |
|
| 244 |
+
|---|---:|---:|---:|---:|
|
| 245 |
+
| GPE | 0.953333 | 0.934641 | 0.943894 | 153 |
|
| 246 |
+
| ORG | 0.964029 | 0.964029 | 0.964029 | 139 |
|
| 247 |
+
| DATE | 0.959677 | 0.991667 | 0.975410 | 120 |
|
| 248 |
+
| PERS | 0.978873 | 0.972028 | 0.975439 | 143 |
|
| 249 |
+
| NORP | 0.960317 | 0.968000 | 0.964143 | 125 |
|
| 250 |
+
| ORDINAL | 0.976190 | 1.000000 | 0.987952 | 82 |
|
| 251 |
+
| OCC | 1.000000 | 0.960000 | 0.979592 | 50 |
|
| 252 |
+
| EVENT | 1.000000 | 0.928571 | 0.962963 | 14 |
|
| 253 |
+
| CARDINAL | 0.979167 | 0.959184 | 0.969072 | 49 |
|
| 254 |
+
| LOC | 0.933333 | 0.933333 | 0.933333 | 15 |
|
| 255 |
+
| WEBSITE | 0.945455 | 1.000000 | 0.971963 | 52 |
|
| 256 |
+
| FAC | 0.888889 | 1.000000 | 0.941176 | 8 |
|
| 257 |
+
| LAW | 1.000000 | 1.000000 | 1.000000 | 8 |
|
| 258 |
+
| TIME | 0.900000 | 0.947368 | 0.923077 | 19 |
|
| 259 |
+
| MONEY | 0.000000 | 0.000000 | 0.000000 | 1 |
|
| 260 |
+
| CURR | 1.000000 | 0.666667 | 0.800000 | 3 |
|
| 261 |
+
| LANGUAGE | 0.666667 | 0.500000 | 0.571429 | 8 |
|
| 262 |
+
| PERCENT | 1.000000 | 1.000000 | 1.000000 | 2 |
|
| 263 |
+
| PRODUCT | 1.000000 | 1.000000 | 1.000000 | 2 |
|
| 264 |
+
| QUANTITY | 1.000000 | 1.000000 | 1.000000 | 1 |
|
| 265 |
+
| UNIT | 0.000000 | 0.000000 | 0.000000 | 0 |
|
| 266 |
+
|
| 267 |
+
## Confusion-matrix observations
|
| 268 |
+
|
| 269 |
+
The strongest remaining confusions are mostly between semantically adjacent
|
| 270 |
+
coarse types or rare labels with limited support.
|
| 271 |
+
|
| 272 |
+
### Top test-overall confusions
|
| 273 |
+
|
| 274 |
+
| Gold label | Predicted label | Count |
|
| 275 |
+
|---|---|---:|
|
| 276 |
+
| ORG | GPE | 36 |
|
| 277 |
+
| GPE | ORG | 21 |
|
| 278 |
+
| LOC | GPE | 11 |
|
| 279 |
+
| FAC | GPE | 9 |
|
| 280 |
+
| EVENT | ORG | 8 |
|
| 281 |
+
| CARDINAL | ORDINAL | 8 |
|
| 282 |
+
| NORP | ORG | 7 |
|
| 283 |
+
| GPE | NORP | 6 |
|
| 284 |
+
| ORG | NORP | 6 |
|
| 285 |
+
| ORG | PERS | 5 |
|
| 286 |
+
| NORP | PERS | 5 |
|
| 287 |
+
| ORDINAL | ORG | 5 |
|
| 288 |
+
|
| 289 |
+
### Top test-unseen-sentence confusions
|
| 290 |
+
|
| 291 |
+
| Gold label | Predicted label | Count |
|
| 292 |
+
|---|---|---:|
|
| 293 |
+
| GPE | ORG | 3 |
|
| 294 |
+
| ORG | GPE | 3 |
|
| 295 |
+
| GPE | WEBSITE | 2 |
|
| 296 |
+
| PERS | GPE | 2 |
|
| 297 |
+
| CARDINAL | ORDINAL | 2 |
|
| 298 |
+
| LANGUAGE | NORP | 2 |
|
| 299 |
+
| GPE | DATE | 1 |
|
| 300 |
+
| GPE | PERS | 1 |
|
| 301 |
+
| GPE | NORP | 1 |
|
| 302 |
+
| GPE | CARDINAL | 1 |
|
| 303 |
+
| GPE | LOC | 1 |
|
| 304 |
+
| ORG | WEBSITE | 1 |
|
| 305 |
+
|
| 306 |
+
## Artifacts included in this repository
|
| 307 |
+
|
| 308 |
+
- `pytorch_model.bin`: best checkpoint model weights
|
| 309 |
+
- `config.json`, `tokenizer.json`, `tokenizer_config.json`, `special_tokens_map.json`
|
| 310 |
+
- `type_predictor_config.json`: task-specific architecture and label config
|
| 311 |
+
- `modeling_type_predictor.py`: custom PyTorch model wrapper
|
| 312 |
+
- `inference.py`: local inference example
|
| 313 |
+
- `metrics/`: validation/test overall, category-specific, and per-class metrics
|
| 314 |
+
- `confusion_matrices/`: validation/test confusion matrices
|
| 315 |
+
- `predictions/`: row-level validation and test predictions
|
| 316 |
+
- `configs/`: architecture, labels, run config, and training arguments
|
| 317 |
+
- `checkpoint_summary.json`, `run_summary.json`, `evaluation_results.json`
|
| 318 |
+
- `checkpoints/latest/`: latest completed checkpoint snapshot
|
| 319 |
|
| 320 |
## Loading
|
| 321 |
|
| 322 |
+
Because this is a small custom wrapper around NeoAraBERT, load the model through
|
| 323 |
+
the included `modeling_type_predictor.py`.
|
| 324 |
|
| 325 |
```python
|
| 326 |
from huggingface_hub import hf_hub_download
|
|
|
|
| 330 |
spec = importlib.util.spec_from_file_location("modeling_type_predictor", source)
|
| 331 |
module = importlib.util.module_from_spec(spec)
|
| 332 |
spec.loader.exec_module(module)
|
| 333 |
+
|
| 334 |
model, tokenizer, config = module.NeoAraBERTTypePredictor.from_pretrained(
|
| 335 |
"U4RASD/TypePredictor"
|
| 336 |
)
|
| 337 |
+
model.eval()
|
| 338 |
```
|
| 339 |
|
| 340 |
+
## Inference example
|
| 341 |
+
|
| 342 |
+
```python
|
| 343 |
+
import torch
|
| 344 |
+
|
| 345 |
+
sentence = "زار أحمد القاهرة أمس."
|
| 346 |
+
entity = "القاهرة"
|
| 347 |
+
start = sentence.index(entity)
|
| 348 |
+
end = start + len(entity)
|
| 349 |
+
|
| 350 |
+
marked = sentence[:start] + " [ENT] " + sentence[start:end] + " [/ENT] " + sentence[end:]
|
| 351 |
+
batch = tokenizer(
|
| 352 |
+
marked,
|
| 353 |
+
return_tensors="pt",
|
| 354 |
+
truncation=True,
|
| 355 |
+
max_length=config["max_length"],
|
| 356 |
+
)
|
| 357 |
+
|
| 358 |
+
with torch.no_grad():
|
| 359 |
+
logits = model(**batch)["logits"]
|
| 360 |
+
probs = torch.softmax(logits, dim=-1)[0]
|
| 361 |
+
|
| 362 |
+
label_id = int(probs.argmax())
|
| 363 |
+
print(config["labels"][label_id], float(probs[label_id]))
|
| 364 |
+
```
|
| 365 |
|
| 366 |
+
The repository also includes `inference.py`, which handles marker-preserving
|
| 367 |
+
entity-centered truncation with the same context-candidate policy used during
|
| 368 |
+
training.
|
| 369 |
|
| 370 |
## Limitations
|
| 371 |
|
| 372 |
+
- This is not a full NER model; it requires a known span.
|
| 373 |
+
- There is no `NONE`/non-entity class in this training run.
|
| 374 |
+
- Overall metrics are not a pure unseen-sentence estimate because the split is
|
| 375 |
+
mention-level and intentionally contains sentence overlap. Use
|
| 376 |
+
`unseen_sentence` metrics for the stricter generalization view.
|
| 377 |
+
- Rare labels such as `UNIT`, `QUANTITY`, `PRODUCT`, `PERCENT`, and `LANGUAGE`
|
| 378 |
+
have much lower support than GPE/ORG/DATE/PERS/NORP.
|
| 379 |
+
- The first locked experiment intentionally avoided imbalance correction, extra
|
| 380 |
+
classifier layers, span pooling, or threshold tuning.
|
| 381 |
+
|
| 382 |
+
## Reproducibility notes
|
| 383 |
+
|
| 384 |
+
The generated artifacts contain the run configuration, training arguments,
|
| 385 |
+
Trainer log history, dataset validation report, encoding validation report,
|
| 386 |
+
metrics, predictions, and confusion matrices. The test set was evaluated only
|
| 387 |
+
after training and checkpoint selection; it was not used for checkpoint or
|
| 388 |
+
hyperparameter selection.
|
| 389 |
+
|
| 390 |
+
Final model card generated from local RunPod artifacts at `2026-07-12T19:05:52+00:00`.
|
release_manifest.json
CHANGED
|
@@ -1,238 +1,238 @@
|
|
| 1 |
{
|
| 2 |
"README.md": {
|
| 3 |
-
"
|
| 4 |
-
"
|
| 5 |
},
|
| 6 |
"checkpoint_summary.json": {
|
| 7 |
-
"
|
| 8 |
-
"
|
| 9 |
},
|
| 10 |
"checkpoints/latest/optimizer.pt": {
|
| 11 |
-
"
|
| 12 |
-
"
|
| 13 |
},
|
| 14 |
"checkpoints/latest/pytorch_model.bin": {
|
| 15 |
-
"
|
| 16 |
-
"
|
| 17 |
},
|
| 18 |
"checkpoints/latest/rng_state.pth": {
|
| 19 |
-
"
|
| 20 |
-
"
|
| 21 |
},
|
| 22 |
"checkpoints/latest/scaler.pt": {
|
| 23 |
-
"
|
| 24 |
-
"
|
| 25 |
},
|
| 26 |
"checkpoints/latest/scheduler.pt": {
|
| 27 |
-
"
|
| 28 |
-
"
|
| 29 |
},
|
| 30 |
"checkpoints/latest/trainer_state.json": {
|
| 31 |
-
"
|
| 32 |
-
"
|
| 33 |
},
|
| 34 |
"checkpoints/latest/training_args.bin": {
|
| 35 |
-
"
|
| 36 |
-
"
|
| 37 |
},
|
| 38 |
"config.json": {
|
| 39 |
-
"
|
| 40 |
-
"
|
| 41 |
},
|
| 42 |
"configs/architecture_config.json": {
|
| 43 |
-
"
|
| 44 |
-
"
|
| 45 |
},
|
| 46 |
"configs/id2label.json": {
|
| 47 |
-
"
|
| 48 |
-
"
|
| 49 |
},
|
| 50 |
"configs/label2id.json": {
|
| 51 |
-
"
|
| 52 |
-
"
|
| 53 |
},
|
| 54 |
"configs/run_config.json": {
|
| 55 |
-
"
|
| 56 |
-
"
|
| 57 |
},
|
| 58 |
"configs/training_arguments.json": {
|
| 59 |
-
"
|
| 60 |
-
"
|
| 61 |
},
|
| 62 |
"confusion_matrices/test_overall_confusion_matrix.csv": {
|
| 63 |
-
"
|
| 64 |
-
"
|
| 65 |
},
|
| 66 |
"confusion_matrices/test_seen_sentence_new_entity_confusion_matrix.csv": {
|
| 67 |
-
"
|
| 68 |
-
"
|
| 69 |
},
|
| 70 |
"confusion_matrices/test_unseen_sentence_confusion_matrix.csv": {
|
| 71 |
-
"
|
| 72 |
-
"
|
| 73 |
},
|
| 74 |
"confusion_matrices/validation_overall_confusion_matrix.csv": {
|
| 75 |
-
"
|
| 76 |
-
"
|
| 77 |
},
|
| 78 |
"confusion_matrices/validation_seen_sentence_new_entity_confusion_matrix.csv": {
|
| 79 |
-
"
|
| 80 |
-
"
|
| 81 |
},
|
| 82 |
"confusion_matrices/validation_unseen_sentence_confusion_matrix.csv": {
|
| 83 |
-
"
|
| 84 |
-
"
|
| 85 |
},
|
| 86 |
"dataset_evaluation_category_counts.csv": {
|
| 87 |
-
"
|
| 88 |
-
"
|
| 89 |
},
|
| 90 |
"dataset_split_type_counts.csv": {
|
| 91 |
-
"
|
| 92 |
-
"
|
| 93 |
},
|
| 94 |
"dataset_validation.json": {
|
| 95 |
-
"
|
| 96 |
-
"
|
| 97 |
},
|
| 98 |
"encoding_summary.json": {
|
| 99 |
-
"
|
| 100 |
-
"
|
| 101 |
},
|
| 102 |
"evaluation_results.json": {
|
| 103 |
-
"
|
| 104 |
-
"
|
| 105 |
},
|
| 106 |
"id2label.json": {
|
| 107 |
-
"
|
| 108 |
-
"
|
| 109 |
},
|
| 110 |
"inference.py": {
|
| 111 |
-
"
|
| 112 |
-
"
|
| 113 |
},
|
| 114 |
"label2id.json": {
|
| 115 |
-
"
|
| 116 |
-
"
|
| 117 |
},
|
| 118 |
"metrics/test_overall.json": {
|
| 119 |
-
"
|
| 120 |
-
"
|
| 121 |
},
|
| 122 |
"metrics/test_overall_per_class.csv": {
|
| 123 |
-
"
|
| 124 |
-
"
|
| 125 |
},
|
| 126 |
"metrics/test_overall_per_class.json": {
|
| 127 |
-
"
|
| 128 |
-
"
|
| 129 |
},
|
| 130 |
"metrics/test_seen_sentence_new_entity.json": {
|
| 131 |
-
"
|
| 132 |
-
"
|
| 133 |
},
|
| 134 |
"metrics/test_seen_sentence_new_entity_per_class.csv": {
|
| 135 |
-
"
|
| 136 |
-
"
|
| 137 |
},
|
| 138 |
"metrics/test_seen_sentence_new_entity_per_class.json": {
|
| 139 |
-
"
|
| 140 |
-
"
|
| 141 |
},
|
| 142 |
"metrics/test_unseen_sentence.json": {
|
| 143 |
-
"
|
| 144 |
-
"
|
| 145 |
},
|
| 146 |
"metrics/test_unseen_sentence_per_class.csv": {
|
| 147 |
-
"
|
| 148 |
-
"
|
| 149 |
},
|
| 150 |
"metrics/test_unseen_sentence_per_class.json": {
|
| 151 |
-
"
|
| 152 |
-
"
|
| 153 |
},
|
| 154 |
"metrics/validation_overall.json": {
|
| 155 |
-
"
|
| 156 |
-
"
|
| 157 |
},
|
| 158 |
"metrics/validation_overall_per_class.csv": {
|
| 159 |
-
"
|
| 160 |
-
"
|
| 161 |
},
|
| 162 |
"metrics/validation_overall_per_class.json": {
|
| 163 |
-
"
|
| 164 |
-
"
|
| 165 |
},
|
| 166 |
"metrics/validation_seen_sentence_new_entity.json": {
|
| 167 |
-
"
|
| 168 |
-
"
|
| 169 |
},
|
| 170 |
"metrics/validation_seen_sentence_new_entity_per_class.csv": {
|
| 171 |
-
"
|
| 172 |
-
"
|
| 173 |
},
|
| 174 |
"metrics/validation_seen_sentence_new_entity_per_class.json": {
|
| 175 |
-
"
|
| 176 |
-
"
|
| 177 |
},
|
| 178 |
"metrics/validation_unseen_sentence.json": {
|
| 179 |
-
"
|
| 180 |
-
"
|
| 181 |
},
|
| 182 |
"metrics/validation_unseen_sentence_per_class.csv": {
|
| 183 |
-
"
|
| 184 |
-
"
|
| 185 |
},
|
| 186 |
"metrics/validation_unseen_sentence_per_class.json": {
|
| 187 |
-
"
|
| 188 |
-
"
|
| 189 |
},
|
| 190 |
"modeling_type_predictor.py": {
|
| 191 |
-
"
|
| 192 |
-
"
|
| 193 |
},
|
| 194 |
"predictions/test_predictions.jsonl": {
|
| 195 |
-
"
|
| 196 |
-
"
|
| 197 |
},
|
| 198 |
"predictions/validation_predictions.jsonl": {
|
| 199 |
-
"
|
| 200 |
-
"
|
| 201 |
},
|
| 202 |
"pytorch_model.bin": {
|
| 203 |
-
"
|
| 204 |
-
"
|
| 205 |
},
|
| 206 |
"run_config.json": {
|
| 207 |
-
"
|
| 208 |
-
"
|
| 209 |
},
|
| 210 |
"run_summary.json": {
|
| 211 |
-
"
|
| 212 |
-
"
|
| 213 |
},
|
| 214 |
"special_tokens_map.json": {
|
| 215 |
-
"
|
| 216 |
-
"
|
| 217 |
},
|
| 218 |
"tokenizer.json": {
|
| 219 |
-
"
|
| 220 |
-
"
|
| 221 |
},
|
| 222 |
"tokenizer_config.json": {
|
| 223 |
-
"
|
| 224 |
-
"
|
| 225 |
},
|
| 226 |
"train_metrics.json": {
|
| 227 |
-
"
|
| 228 |
-
"
|
| 229 |
},
|
| 230 |
"trainer_log_history.json": {
|
| 231 |
-
"
|
| 232 |
-
"
|
| 233 |
},
|
| 234 |
"type_predictor_config.json": {
|
| 235 |
-
"
|
| 236 |
-
"
|
| 237 |
}
|
| 238 |
-
}
|
|
|
|
| 1 |
{
|
| 2 |
"README.md": {
|
| 3 |
+
"sha256": "c7e6e91bef30977ce7262965457cc1ecec2fd2bc9ae73135e6f3df46837da776",
|
| 4 |
+
"bytes": 13464
|
| 5 |
},
|
| 6 |
"checkpoint_summary.json": {
|
| 7 |
+
"sha256": "eab687212a3bd05b0659ff7a255dc552b7d5bca87f75e5beffd318e5f384bf2b",
|
| 8 |
+
"bytes": 570
|
| 9 |
},
|
| 10 |
"checkpoints/latest/optimizer.pt": {
|
| 11 |
+
"sha256": "f50dc9dafa68f1d1e36c152f307cd243c56b3f4a76f1e8a88bbaf64fe8becbf6",
|
| 12 |
+
"bytes": 1985443322
|
| 13 |
},
|
| 14 |
"checkpoints/latest/pytorch_model.bin": {
|
| 15 |
+
"sha256": "a3ef5bb9771e8705208a0f3e97788305b09b19d0be51fcf939f736fc5c13fb72",
|
| 16 |
+
"bytes": 992721978
|
| 17 |
},
|
| 18 |
"checkpoints/latest/rng_state.pth": {
|
| 19 |
+
"sha256": "923e5b6a51442e1617111f28197dff670cd469df9e1de7d6216bb14030444691",
|
| 20 |
+
"bytes": 14244
|
| 21 |
},
|
| 22 |
"checkpoints/latest/scaler.pt": {
|
| 23 |
+
"sha256": "9cfb01e00465a5891d4570d1c7db65c31941253c42933e7c82e41ca335ed13f5",
|
| 24 |
+
"bytes": 988
|
| 25 |
},
|
| 26 |
"checkpoints/latest/scheduler.pt": {
|
| 27 |
+
"sha256": "caa8aa904b6ba57c95fdeb6646034707f9aa8f50f77bbe64ebde1dfff45a23e3",
|
| 28 |
+
"bytes": 1064
|
| 29 |
},
|
| 30 |
"checkpoints/latest/trainer_state.json": {
|
| 31 |
+
"sha256": "77492f6ad4077e9b12c0571c762371cb7021036729d872a64f31ec7052bb0937",
|
| 32 |
+
"bytes": 123247
|
| 33 |
},
|
| 34 |
"checkpoints/latest/training_args.bin": {
|
| 35 |
+
"sha256": "c5b6cfce7351bff5319a51e160874bfbfcfd153324040fb99c38f5fff0732e17",
|
| 36 |
+
"bytes": 5368
|
| 37 |
},
|
| 38 |
"config.json": {
|
| 39 |
+
"sha256": "463c0a301c7b3b54361ea09146bfbd6292d3aa64a71717a08574106f5e1ce746",
|
| 40 |
+
"bytes": 1709
|
| 41 |
},
|
| 42 |
"configs/architecture_config.json": {
|
| 43 |
+
"sha256": "6393a8ca6ac080c913613b8576716a061d265478bfcc6c2bf8de915f00b18280",
|
| 44 |
+
"bytes": 845
|
| 45 |
},
|
| 46 |
"configs/id2label.json": {
|
| 47 |
+
"sha256": "503df762ab4b43800e4aeb9e69f6bc8d2c72db46f5ec794ffec2335ba8bdd2ad",
|
| 48 |
+
"bytes": 348
|
| 49 |
},
|
| 50 |
"configs/label2id.json": {
|
| 51 |
+
"sha256": "908a58f67dafe45b311c4ae6f1adf8d6f90f6c5e1a28d3ba638306d59f7415dc",
|
| 52 |
+
"bytes": 306
|
| 53 |
},
|
| 54 |
"configs/run_config.json": {
|
| 55 |
+
"sha256": "1b02b8cf7d96c3ef6e665e5df32150a763ce3f90d1252a107bc60de2151d90b5",
|
| 56 |
+
"bytes": 2206
|
| 57 |
},
|
| 58 |
"configs/training_arguments.json": {
|
| 59 |
+
"sha256": "effe18203c830e0c72cf95698dba5be8bb9789131fc0f0f3689df9bd3bf53d14",
|
| 60 |
+
"bytes": 4143
|
| 61 |
},
|
| 62 |
"confusion_matrices/test_overall_confusion_matrix.csv": {
|
| 63 |
+
"sha256": "a6a3478a326167535467ee193f80087ebcddd7ddad7f343ea20844918b27938e",
|
| 64 |
+
"bytes": 1176
|
| 65 |
},
|
| 66 |
"confusion_matrices/test_seen_sentence_new_entity_confusion_matrix.csv": {
|
| 67 |
+
"sha256": "e3eb8a6fcdab81a430124c5eb0429cde3e2a53e9af154d5e7f7c99e047b07f80",
|
| 68 |
+
"bytes": 1173
|
| 69 |
},
|
| 70 |
"confusion_matrices/test_unseen_sentence_confusion_matrix.csv": {
|
| 71 |
+
"sha256": "ecc3122d832647ee403b6c496ea778e112aa23bb500ad5eeeb7f56fab8a5cf86",
|
| 72 |
+
"bytes": 1153
|
| 73 |
},
|
| 74 |
"confusion_matrices/validation_overall_confusion_matrix.csv": {
|
| 75 |
+
"sha256": "5114c46fc0b115983cb80f6ea529b9e84160c374aa94bebea16df2bc2f3d405c",
|
| 76 |
+
"bytes": 1177
|
| 77 |
},
|
| 78 |
"confusion_matrices/validation_seen_sentence_new_entity_confusion_matrix.csv": {
|
| 79 |
+
"sha256": "0444cb2c92b59f1bed7018b2c47513edabe399c111fbc7efc9ca2b5b8081edd2",
|
| 80 |
+
"bytes": 1174
|
| 81 |
},
|
| 82 |
"confusion_matrices/validation_unseen_sentence_confusion_matrix.csv": {
|
| 83 |
+
"sha256": "6dc3a930fcde8186ccd0a610e9ef7415f467d56384a197c297d0b5b7fb07c8d7",
|
| 84 |
+
"bytes": 1152
|
| 85 |
},
|
| 86 |
"dataset_evaluation_category_counts.csv": {
|
| 87 |
+
"sha256": "43f32fbb579a5c794e6b816002e7469840bd1a8ec8ccebc20af9b47bc5ec0d08",
|
| 88 |
+
"bytes": 3206
|
| 89 |
},
|
| 90 |
"dataset_split_type_counts.csv": {
|
| 91 |
+
"sha256": "320dd465824ff591d1cd8d0a58bf2dd82fe3ea641e05e5f0b20260d17caa7e04",
|
| 92 |
+
"bytes": 1697
|
| 93 |
},
|
| 94 |
"dataset_validation.json": {
|
| 95 |
+
"sha256": "4bd8ce139d0750f4bd28eb5f3372493a80e895f8e73e169f53e74a3ad675f10b",
|
| 96 |
+
"bytes": 19049
|
| 97 |
},
|
| 98 |
"encoding_summary.json": {
|
| 99 |
+
"sha256": "bdf18ca5a72e07653af51f47dd72f808df491d52f4e0e51550c500cc9151ade8",
|
| 100 |
+
"bytes": 1249
|
| 101 |
},
|
| 102 |
"evaluation_results.json": {
|
| 103 |
+
"sha256": "8f835bb1e5a65a5154d85a1645aaed903abf7777629bb51cec9c930b186a2825",
|
| 104 |
+
"bytes": 9808
|
| 105 |
},
|
| 106 |
"id2label.json": {
|
| 107 |
+
"sha256": "503df762ab4b43800e4aeb9e69f6bc8d2c72db46f5ec794ffec2335ba8bdd2ad",
|
| 108 |
+
"bytes": 348
|
| 109 |
},
|
| 110 |
"inference.py": {
|
| 111 |
+
"sha256": "55edf36956dbdae52144ab2d1f60017c6e2ca4e8a752e183e6e0770541a254d4",
|
| 112 |
+
"bytes": 2900
|
| 113 |
},
|
| 114 |
"label2id.json": {
|
| 115 |
+
"sha256": "908a58f67dafe45b311c4ae6f1adf8d6f90f6c5e1a28d3ba638306d59f7415dc",
|
| 116 |
+
"bytes": 306
|
| 117 |
},
|
| 118 |
"metrics/test_overall.json": {
|
| 119 |
+
"sha256": "e77112678308e6757a714562efebe51c6930e2148c64ce685a41350036aa5c77",
|
| 120 |
+
"bytes": 1061
|
| 121 |
},
|
| 122 |
"metrics/test_overall_per_class.csv": {
|
| 123 |
+
"sha256": "d4cef72641eccc133d3a5ccc0c65bdb8dd659e7266f7adbef3cdb710513a4419",
|
| 124 |
+
"bytes": 1554
|
| 125 |
},
|
| 126 |
"metrics/test_overall_per_class.json": {
|
| 127 |
+
"sha256": "604f78dc0fcdaca864ad0f1ad78bb30df26b039ed548aebd03a5df98b1179dbd",
|
| 128 |
+
"bytes": 3238
|
| 129 |
},
|
| 130 |
"metrics/test_seen_sentence_new_entity.json": {
|
| 131 |
+
"sha256": "4928c87da8621657f325b15bf0bcfc8fffec1478e80bd429c54f2811357860ad",
|
| 132 |
+
"bytes": 1076
|
| 133 |
},
|
| 134 |
"metrics/test_seen_sentence_new_entity_per_class.csv": {
|
| 135 |
+
"sha256": "6818610a21e9f5e625a3001cc1a958a41380bcafa88319c176c66c28f03e934d",
|
| 136 |
+
"bytes": 1549
|
| 137 |
},
|
| 138 |
"metrics/test_seen_sentence_new_entity_per_class.json": {
|
| 139 |
+
"sha256": "ed4e05f2c336a520ac32c58e76265b1b9171906134e42c5f2575d8717e46f88d",
|
| 140 |
+
"bytes": 3233
|
| 141 |
},
|
| 142 |
"metrics/test_unseen_sentence.json": {
|
| 143 |
+
"sha256": "39bb689931b416e73e5f14fd5602014d2f6bbf77be3f65ff03f69c9fc4737e55",
|
| 144 |
+
"bytes": 1054
|
| 145 |
},
|
| 146 |
"metrics/test_unseen_sentence_per_class.csv": {
|
| 147 |
+
"sha256": "4383fb8ee69536fecdc198d3f37067bf4f4c6f1f72e76fa7c0cb6a6930b332bb",
|
| 148 |
+
"bytes": 1261
|
| 149 |
},
|
| 150 |
"metrics/test_unseen_sentence_per_class.json": {
|
| 151 |
+
"sha256": "b7c90a96eda66ccff26fb551e12f3aa7363ac35c1b6f04cc80f9d7e05277b52b",
|
| 152 |
+
"bytes": 2945
|
| 153 |
},
|
| 154 |
"metrics/validation_overall.json": {
|
| 155 |
+
"sha256": "497f8980c7a7a381a14fd5d428e2bdffb3340d822246016ecd18e36d0872d6a3",
|
| 156 |
+
"bytes": 1066
|
| 157 |
},
|
| 158 |
"metrics/validation_overall_per_class.csv": {
|
| 159 |
+
"sha256": "3ed9e3aafac9bbadd579d6be956fd6ef67fe2a6e9eb0790243953e021eee8a0d",
|
| 160 |
+
"bytes": 1550
|
| 161 |
},
|
| 162 |
"metrics/validation_overall_per_class.json": {
|
| 163 |
+
"sha256": "fb5d5dbdf95860bb27e4c3623c90085f7dcf17f7a396297e27d27c2a2db9842e",
|
| 164 |
+
"bytes": 3234
|
| 165 |
},
|
| 166 |
"metrics/validation_seen_sentence_new_entity.json": {
|
| 167 |
+
"sha256": "ce85076cdb803d4e9d63297f9c27e8679324ebd099fc7afd334898106129862c",
|
| 168 |
+
"bytes": 1081
|
| 169 |
},
|
| 170 |
"metrics/validation_seen_sentence_new_entity_per_class.csv": {
|
| 171 |
+
"sha256": "81f77dae89ac5fca28e460a19d27135c827ec3622e00293f718354f6e47c6ffb",
|
| 172 |
+
"bytes": 1528
|
| 173 |
},
|
| 174 |
"metrics/validation_seen_sentence_new_entity_per_class.json": {
|
| 175 |
+
"sha256": "a4ec736db8d8e255f9b76c9ce53d03684355ea9d7d792c747311e3548c4aac3f",
|
| 176 |
+
"bytes": 3212
|
| 177 |
},
|
| 178 |
"metrics/validation_unseen_sentence.json": {
|
| 179 |
+
"sha256": "291a980ee38aecbde1588d19fb0a411f04cca0699908f6023f92d5613fe72497",
|
| 180 |
+
"bytes": 1056
|
| 181 |
},
|
| 182 |
"metrics/validation_unseen_sentence_per_class.csv": {
|
| 183 |
+
"sha256": "ad60bbeff3ae6b2d2ba3d81c6f827a2173b82bfbf3d41910264529e0c6462c8a",
|
| 184 |
+
"bytes": 1310
|
| 185 |
},
|
| 186 |
"metrics/validation_unseen_sentence_per_class.json": {
|
| 187 |
+
"sha256": "1478697313a908715ba8848a10657e9d1c48136ee36d80b6901e100d783c5fc2",
|
| 188 |
+
"bytes": 2994
|
| 189 |
},
|
| 190 |
"modeling_type_predictor.py": {
|
| 191 |
+
"sha256": "41a6c1b1ac7615622b8959fb7bf0561be1930b4b63a6346ba89eec89d40baef8",
|
| 192 |
+
"bytes": 4316
|
| 193 |
},
|
| 194 |
"predictions/test_predictions.jsonl": {
|
| 195 |
+
"sha256": "fcf7973d1e70244cde5b8639296e0150392d176f3150037dadfe01ddeaa88a54",
|
| 196 |
+
"bytes": 11819131
|
| 197 |
},
|
| 198 |
"predictions/validation_predictions.jsonl": {
|
| 199 |
+
"sha256": "588255cb6dedcdfdf9945b8c92ab1f8cf2f04a5319c38766393bfa5878820a6b",
|
| 200 |
+
"bytes": 11667880
|
| 201 |
},
|
| 202 |
"pytorch_model.bin": {
|
| 203 |
+
"sha256": "3415bde16ad0aef69806e0effcd73150f82543544f22be1f7680d685ddb59081",
|
| 204 |
+
"bytes": 992721978
|
| 205 |
},
|
| 206 |
"run_config.json": {
|
| 207 |
+
"sha256": "1b02b8cf7d96c3ef6e665e5df32150a763ce3f90d1252a107bc60de2151d90b5",
|
| 208 |
+
"bytes": 2206
|
| 209 |
},
|
| 210 |
"run_summary.json": {
|
| 211 |
+
"sha256": "8f835bb1e5a65a5154d85a1645aaed903abf7777629bb51cec9c930b186a2825",
|
| 212 |
+
"bytes": 9808
|
| 213 |
},
|
| 214 |
"special_tokens_map.json": {
|
| 215 |
+
"sha256": "73794eb3633b96a848db5b43fdad76d91200070731312aa472edb781e9a344f2",
|
| 216 |
+
"bytes": 1011
|
| 217 |
},
|
| 218 |
"tokenizer.json": {
|
| 219 |
+
"sha256": "a6fa2f14546d2be1ed6b551c2bcb7ecf220e757b59037dc2d112295c11924c14",
|
| 220 |
+
"bytes": 1706886
|
| 221 |
},
|
| 222 |
"tokenizer_config.json": {
|
| 223 |
+
"sha256": "7ca6df45cf310081b97b1b43b54316b7b380bbd4bbfe39966b24d1487359584d",
|
| 224 |
+
"bytes": 1995
|
| 225 |
},
|
| 226 |
"train_metrics.json": {
|
| 227 |
+
"sha256": "21b773839d7546ec69dcd4d52ed33e3424ccd05bb1cb7b198f5f9c193f904e1a",
|
| 228 |
+
"bytes": 195
|
| 229 |
},
|
| 230 |
"trainer_log_history.json": {
|
| 231 |
+
"sha256": "68dd9635c6a2cc8e0b2dd3f491d40f8a5d440b50b4490024575b5b67ce90e75d",
|
| 232 |
+
"bytes": 113832
|
| 233 |
},
|
| 234 |
"type_predictor_config.json": {
|
| 235 |
+
"sha256": "6393a8ca6ac080c913613b8576716a061d265478bfcc6c2bf8de915f00b18280",
|
| 236 |
+
"bytes": 845
|
| 237 |
}
|
| 238 |
+
}
|