CoolFace
Modelpublic

yonad2008/predict_xgb_phi4_14b

sourceHugging Faceupdated 5mo agoView on Hugging Face
0likes
Model Card

XGBoost Jailbreak Prediction Model: phi4:14b

XGBoost + TF-IDF (+ optional TruncatedSVD) classifier for unsafe/jailbreak likelihood in multi-turn conversations.

Evaluation Results (best fold: 1)

MetricValue
F10.2807
PR-AUC0.2896
ROC-AUC0.7231
Precision0.2500
Recall0.3200
Best Threshold0.20

Training Details

  • Target model: phi4:14b
  • Datasets: harmful_behaviors
  • K-Folds: 5
  • Input format: single turn: category + strategy_name + one TURN line
  • TF-IDF ngram_range: (1, 1)
  • TF-IDF max_features: 120000
  • TruncatedSVD: enabled True, requested n_components=1024
  • XGBoost n_estimators: 971
  • XGBoost learning_rate: 0.045325359791945935
  • XGBoost max_depth: 7

Dataset Size (training samples)

Prepared turn-level samples: 1611 (unsafe: 119, safe: 1492)