
# Jailbreak Classifier

Classifies prompts as jailbreaks or benign. This is a fine-tune checkpoint of [bert-base-uncased](https://huggingface.co/bert-base-uncased) on the [jailbreak-classification](https://huggingface.co/datasets/jackhhao/jailbreak-classification) dataaset.

## Training Details

### Training Data

Fine-tuned on the [jailbreak-classification](https://huggingface.co/datasets/jackhhao/jailbreak-classification) dataaset.

### Training Procedure 

#### Training Hyperparameters

Fine-tuning hyper-parameters:
- learning_rate = 5e-5
- per_device_train_batch_size = 8
- per_device_eval_batch_size = 8
- lr_scheduler_type = linear
- num_train_epochs = 5.0
{"language": ["en"], "license": "apache-2.0", "library_name": "transformers", "tags": ["jailbreak", "security", "moderation"], "datasets": ["Open-Orca/OpenOrca", "jackhhao/jailbreak-classification"], "metrics": ["accuracy"], "pipeline_tag": "text-classification"}
transformers pytorch bert text-classification jailbreak security moderation en dataset:Open-Orca/OpenOrca dataset:jackhhao/jailbreak-classification license:apache-2.0 autotrain_compatible endpoints_compatible region:us