{
  "label_to_id": {
    "safe": 0,
    "unsafe": 1
  },
  "violated_category_to_id": {
    "Controlled/Regulated Substances": 0,
    "Copyright/Trademark/Plagiarism": 1,
    "Criminal Planning/Confessions": 2,
    "Fraud/Deception": 3,
    "Guns and Illegal Weapons": 4,
    "Harassment": 5,
    "Hate/Identity Hate": 6,
    "High Risk Gov Decision Making": 7,
    "Illegal Activity": 8,
    "Immoral/Unethical": 9,
    "Malware": 10,
    "Manipulation": 11,
    "Needs Caution": 12,
    "Other": 13,
    "PII/Privacy": 14,
    "Political/Misinformation/Conspiracy": 15,
    "Profanity": 16,
    "Sexual": 17,
    "Sexual (minor)": 18,
    "Suicide and Self Harm": 19,
    "Threat": 20,
    "Unauthorized Advice": 21,
    "Violence": 22
  },
  "id_to_label": {
    "0": "safe",
    "1": "unsafe"
  },
  "id_to_violated_category": {
    "0": "Controlled/Regulated Substances",
    "1": "Copyright/Trademark/Plagiarism",
    "2": "Criminal Planning/Confessions",
    "3": "Fraud/Deception",
    "4": "Guns and Illegal Weapons",
    "5": "Harassment",
    "6": "Hate/Identity Hate",
    "7": "High Risk Gov Decision Making",
    "8": "Illegal Activity",
    "9": "Immoral/Unethical",
    "10": "Malware",
    "11": "Manipulation",
    "12": "Needs Caution",
    "13": "Other",
    "14": "PII/Privacy",
    "15": "Political/Misinformation/Conspiracy",
    "16": "Profanity",
    "17": "Sexual",
    "18": "Sexual (minor)",
    "19": "Suicide and Self Harm",
    "20": "Threat",
    "21": "Unauthorized Advice",
    "22": "Violence"
  },
  "num_prompt_labels": 2,
  "num_response_labels": 2,
  "num_categories": 23,
  "training_config": {
    "sample_percentage": 1.0,
    "max_length": 256,
    "train_batch_size": 16,
    "eval_batch_size": 32,
    "num_epochs": 3,
    "grad_accumulation": 2,
    "num_workers": 2,
    "early_stopping_patience": 2
  },
  "training_type": "FULL_DATASET"
}