Instructions to use whr778/gliner2-base-v1-docee with libraries, inference providers, notebooks, and local apps. Follow these links to get started.
- Libraries
- GLiNER2
How to use whr778/gliner2-base-v1-docee with GLiNER2:
from gliner2 import GLiNER2 model = GLiNER2.from_pretrained("whr778/gliner2-base-v1-docee") # Extract entities text = "Apple CEO Tim Cook announced iPhone 15 in Cupertino yesterday." result = extractor.extract_entities(text, ["company", "person", "product", "location"]) print(result) - Notebooks
- Google Colab
- Kaggle
gliner2-base-v1_docee
A GLiNER2 multi-task information-extraction model (entities, relations, events, and classification) fine-tuned from fastino/gliner2-base-v1.
β οΈ License at a glance
- Effective license: Unverified β review required
- Commercial use: Unverified
- All dataset licenses verified: No
See License for the full determination and per-dataset terms.
Model details
- Base model:
fastino/gliner2-base-v1 - Library:
gliner2 - Tasks: entity, relation, event, and classification extraction
- Experiment:
gliner2-base-v1_docee
Training data
1 dataset used for this run. 21,966 training records (val: 2,748, test: 2,771).
| Dataset | Task(s) | Train | Val | Test | Language | License | Source |
|---|---|---|---|---|---|---|---|
| DocEE | Event extraction (doc-level) | 21,966 | 2,748 | 2,771 | en | see source | link |
Dataset notes
- DocEE β Document-level event extraction; one event per document mapped to role-typed entities and document classification.
Training procedure
| Setting | Value |
|---|---|
| Trained on | 2026-07-19 |
| Duration | 17h 42m |
| Throughput | 14.3 samples/s |
| Epochs | 15 |
| Batch size | 16 (Γ 2 grad-accum) |
| Encoder LR | 1e-05 |
| Task-head LR | 0.0003 |
| Weight decay | 0.01 |
| Scheduler | cosine_restarts (warmup 0.05) |
| Precision | bf16 |
| Max grad norm | 1.0 |
| Best-checkpoint metric | eval_classification_strict_micro_f1 |
| Seed | 42 |
Evaluation
Decision threshold: 0.5 (config default).
Blind test (held-out test splits)
Micro precision / recall / F1, strict β relaxed.
| Category | Precision | Recall | F1 | Support |
|---|---|---|---|---|
| entity | 0.257 β 0.416 | 0.473 β 0.765 | 0.333 β 0.539 | 19188 |
| classification | 0.993 β 0.996 | 0.993 β 0.996 | 0.993 β 0.996 | 2771 |
Best checkpoint (validation)
Micro precision / recall / F1, strict β relaxed.
| Category | Precision | Recall | F1 | Support |
|---|---|---|---|---|
| entity | 0.438 β 0.666 | 0.456 β 0.695 | 0.447 β 0.680 | 27078 |
| classification | 0.906 β 0.924 | 0.906 β 0.924 | 0.906 β 0.924 | 8093 |
License
Effective license: Unverified β review required. This model is a derivative of its base model and every training dataset, so the most restrictive term across all of them governs the whole model.
- Commercial use: Unverified
- Share-alike obligation: No
- All licenses verified: No
- Base model: gliner2-base-v1 β see model card
Unverified β verify the upstream terms before redistribution
- DocEE (see source)
- gliner2-base-v1 (see model card)
License strings are copied verbatim from each dataset's card/source and from
tools/train/dataset_registry.yaml. "see card"/"see source"/"other" mean the upstream declares no clear license β treat as unverified. This summary is informational, not legal advice; confirm terms before redistribution or commercial use.
Citation
If you use this model, please cite GLiNER2 and the underlying datasets (linked in Training data).
Model card generated automatically at the end of training (2026-07-19).
- Downloads last month
- 2
Model tree for whr778/gliner2-base-v1-docee
Base model
fastino/gliner2-base-v1