predict_gnn_phi4_14b / metadata.json
yonad2008's picture
Upload GNN turn-level model artifacts
2c69dbe verified
raw
history blame contribute delete
806 Bytes
{
"csv": "/home/digayona/multi_turn_jailbreak_RL/GNN/turns_table_llama3_8b_harmbench.csv",
"target_model": "phi4:14b",
"threshold": 0.27,
"sentence_model_name": "sentence-transformers/all-MiniLM-L6-v2",
"n_rows": 707,
"n_models": 1,
"split_col": "goal",
"seed": 42,
"turn_norm_mode": "dataset_max",
"turn_norm_denom": 22.0,
"session_len_norm_mode": "dataset_max",
"session_len_norm_denom": 12.0,
"model_kwargs": {
"hidden_channels": 128,
"num_layers": 2,
"dropout": 0.3,
"stage_aware_head": false,
"stage_bucket_start": null
},
"use_turn_bucket_features": false,
"test_metrics": {
"roc_auc": 0.9592919442561716,
"pr_auc": 0.9782228704301538,
"f1": 0.941144695354281,
"precision": 0.9681518630856448,
"recall": 0.9162547510892741
}
}