This project is built for detecting AI generated text using the BERT family of models. You can found the training details in the notebook. If you wish to change the training parameters with ease, you can choose to open the notebook with Google Colab, which allows you to alter the parameters in a handy way.
A few highlights on what the notebook has done:
- Training of BERT or RoBERTa model with the Wikipedia Intro dataset
- Achieved 99% accuracy in differentiating between human/GPT written wiki intro
- Interactive interface for changing training parameters, you can choose to train it in BERT or RoBERTa
Enjoy! :)