File size: 1,089 Bytes
5612646 58d880f 5612646 58d880f 5612646 58d880f 5612646 58d880f 5612646 58d880f 5612646 58d880f 5612646 68d73db 58d880f e8a03fc 68d73db e8a03fc |
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 39 40 41 42 43 44 |
---
language:
- en
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
Second fine-tuning hyper-parameters(on train(0.8) and val(0.2))
- learning_rate = 5e-5
- train_batch_size = 8
- eval_batch_size = 8
- lr_scheduler_type = linear
- num_train_epochs = 5.0
Fecond fine-tuning hyper-parameters(on train and test)
- learning_rate = 1e-5
- train_batch_size = 8
- eval_batch_size = 8
- lr_scheduler_type = linear
- num_train_epochs = 3.0 |