| { | |
| "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" | |
| } |