File size: 1,323 Bytes
46cc63a
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
e317d56
 
46cc63a
e317d56
46cc63a
 
 
 
e317d56
 
46cc63a
e317d56
46cc63a
e317d56
46cc63a
e317d56
46cc63a
 
 
e317d56
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
"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"