metadata
license: mit
base_model:
- prajjwal1/bert-tiny
pipeline_tag: text-classification
library_name: transformers
BERT-tiny RAID Detector
This model is used to detect AI-generated texts from the RAID dataset. The model outputs a score corresponding to the likelihood that the text is human.
This model was trained on a stratified subset of RAID's training data.
Model Details
- Base Model:
prajjwal1/bert-tiny - Optimizer:
Adam - Loss Function:
FocalLoss - Hyperparameters:
- Learning rate:
5e-5 - Epochs: 5
- Learning rate:
1e-5 - Epochs: 5
- Learning rate: