predict_gnn_gemma4_e4b / metadata.json
yonad2008's picture
Upload GNN turn-level model artifacts
6b8cf7b verified
{
"csv": "/home/digayona/multi_turn_jailbreak_RL/GNN/turns_table_llama3_8b_harmbench.csv",
"target_model": "gemma4:e4b",
"threshold": 0.75,
"sentence_model_name": "sentence-transformers/all-MiniLM-L6-v2",
"n_rows": 425,
"n_models": 1,
"split_col": "goal",
"seed": 42,
"model_kwargs": {
"hidden_channels": 128,
"num_layers": 2,
"dropout": 0.3
},
"test_metrics": {
"roc_auc": 0.9512508016674683,
"pr_auc": 0.9358038660808109,
"f1": 0.8936682510495091,
"precision": 0.9589473684210527,
"recall": 0.8378787878787879
}
}