CoolFace
Modelpublic

rafmacalaba/gliner-datause-catchall-singlepass

sourceHugging Faceapache-2.0updated 14d agoView on Hugging Face
0likes16downloads
Model Card

gliner-datause-catchall-singlepass

Single-pass catch-all cascade in one bundle: the rafmacalaba/gliner-datause-mentions-catch-all encoder (frozen, byte-identical) plus an inference-native probe head (probe_head.pt, trained by outputs/gliner-datause-catchall-infer-probe with --feature-source infer). One forward per doc yields proposals (GLiNER DATAMENTION @ 0.1) and keep/drop (`probescore) together — see training/singlepass_infer.py`.

  • head: inputdim 2048, hidden 256, contextradius 64
  • keep knob: probe_score >= thr (global best 0.5 @ F1 0.8467)

Per-origin best-F1 thresholds

originthrf1
fcv_pads_east_africa0.50.8046
general_prwp0.40.8796
jad_paddy_docs0.10.9693
jdc_operational0.60.8019
refugee_pads0.50.8387
reliefweb0.40.7738

Inference

python
from training.singlepass_infer import load_bundle, predict_keep_drop
model, head, bundle = load_bundle('rafmacalaba/gliner-datause-catchall-singlepass', 'rafmacalaba/gliner-datause-catchall-singlepass', 'cuda')
rows = predict_keep_drop(texts, model, head, bundle,
                         propose_thr=0.1, keep_thr=0.3)

Files: pytorch_model.bin + gliner_config.json (encoder), probe_head.pt + head_config.json (head), thresholds.json (operating points), holdout_metrics.json (sweep).