CoolFace
Modelpublic

Olumorothy/ai-human-detector-models

sourceHugging Facemitupdated 3mo agoView on Hugging Face
0likes2downloads
Model Card

AI vs. Human Text Detector — Model Collection

Trained models for detecting AI-generated vs. human-written text. Used by the AI vs. Human Text Detection Space.

Contents

This repository holds the trained artifacts consumed by the app:

  • —roberta_ft/ — fine-tuned `roberta-base` classifier (the primary model)
  • —svm_model.pkl, decision_tree_model.pkl, adaboost_model.pkl — classical ML models
  • —fnn_model.h5, lstm_model.keras, cnn_model.keras — deep-learning models
  • —tfidf_vectorizer.pkl, text_vectorizer_vocab.pkl — feature transformers

Fine-tuned RoBERTa

FacebookAI/roberta-base fine-tuned as a binary classifier (0 = human, 1 = AI).

Hyperparameters

SettingValue
Base modelFacebookAI/roberta-base
Learning rate2e-5
Train / eval batch size8
Gradient accumulation2 (effective batch 16)
Epochs3
Weight decay0.01
Max sequence length512

Results

MetricScore
Train Loss0.4943
Eval Loss0.2022
Eval F10.9988
Validation Accuracy0.9988
Test Accuracy0.9976
Test F10.9976

Training data

~8,000 balanced human/AI text samples, stratified into train/validation/test splits.

Intended use & limitations

Intended for research and educational demonstration of AI-text detection. Predictions are probabilistic estimates, not proof of authorship. Do not use as the sole basis for academic-integrity or disciplinary decisions — accuracy degrades on short texts, edited AI output, and formal human writing.