File size: 902 Bytes
ba0a3ff
dc01ec4
 
ba0a3ff
dc01ec4
 
 
 
 
 
 
 
 
 
 
ba0a3ff
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
---
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
---

# 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