Team Ai
Modelpublic

ShantanuT01/BERT-tiny-RAID

sourceHugging Facemitupdated 1y agoView on Hugging Face
2likes13kdownloads
Model Card

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