| | --- |
| | 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 |
