SignalMod / configs /model_catalog.yaml
Mirae Kang
feat: implement new models and improve UI, #23
46cc63a
Raw
History Blame Contribute Delete
1.32 kB
"Meta-Feature Stacking (Production)":
type: meta_stack
icon: "πŸ†"
description: "Production model β€” frozen Toxic-BERT CLS + metadata + regularized LR meta-learner (Notebook 14)."
speed: "~400ms CPU (first load downloads BERT)"
accuracy: "F1 0.805"
train_test_gap_pp: 2.54
recommended_threshold: 0.381
# display_banner: "Currently using: Meta-Feature Stacking (F1: 0.805, Gap: 2.54%)"
model_path: models/production_final/meta_stack_final.joblib
manifest_path: models/production_final/manifest.json
frozen_bert_id: unitary/toxic-bert
requires: "uv sync --extra hf; models/production_final/meta_stack_final.joblib"
production_default: true
"LR + TF-IDF (Baseline)":
type: local
icon: "⚑"
description: "Esencial sklearn baseline β€” Optuna-tuned logistic regression on TF-IDF (Notebooks 01–03)."
speed: "< 50ms"
accuracy: "F1 0.758"
train_test_gap_pp: 4.76
recommended_threshold: 0.5
model_path: models/baseline/lr_tfidf.joblib
requires: "joblib only"
"Frozen Toxic-BERT (Baseline)":
type: hf_remote
icon: "🧊"
model_id: unitary/toxic-bert
description: "Frozen pretrained Toxic-BERT inference only (Notebook 12 golden baseline)."
speed: "~400ms CPU"
accuracy: "F1 0.790"
train_test_gap_pp: 0.16
recommended_threshold: 0.12
requires: "uv sync --extra hf"