File size: 897 Bytes
ba0a3ff
dc01ec4
 
ba0a3ff
dc01ec4
 
 
 
 
 
 
 
 
 
 
771aa6f
ba0a3ff
dc01ec4
 
 
771aa6f
dc01ec4
 
 
 
 
771aa6f
dc01ec4
 
 
 
 
 
 
771aa6f
 
dc01ec4
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
---
language:
- en
license: apache-2.0
datasets:
- Open-Orca/OpenOrca
- jackhhao/jailbreak-classification
metrics:
- accuracy
library_name: transformers
pipeline_tag: text-classification
tags:
- jailbreak
- security
- moderation
- prompt-injection
---

# 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) dataset.

## Training Details

### Training Data

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

### Training Procedure 

#### Training Hyperparameters

Fine-tuning hyper-parameters:
- learning_rate = 5e-5
- train_batch_size = 8
- eval_batch_size = 8
- lr_scheduler_type = linear
- num_train_epochs = 5.0