r2911/lm_p1
09
๐ง AI Text Detector โ DeBERTa v3 Large (Fine-tuned on Human vs AI Text)
This model is fine-tuned on a labeled dataset for AI-generated vs. Human-written text detection.
โ๏ธ Model Details
- Base Model:
microsoft/deberta-v3-large - Fine-tuned by: @abhinav
- Epochs: 4
- Learning Rate: 2e-05
- Batch Size: 8
- GPU: 80 GB A100
- Optimizer: AdamW (Fused)
- Scheduler: Cosine
- Mixed Precision: FP16
- Gradient Checkpointing: Enabled
๐ Evaluation Results (Test Set)
๐งฎ Confusion Matrix
๐ Example Inference
from transformers import AutoTokenizer, AutoModelForSequenceClassification
import torch
tokenizer = AutoTokenizer.from_pretrained("abhi099k/ai-text-detector-deberta-v3-large-h1")
model = AutoModelForSequenceClassification.from_pretrained("abhi099k/ai-text-detector-deberta-v3-large-h1")
text = "This text was likely written by an AI model."
inputs = tokenizer(text, return_tensors="pt")
outputs = model(**inputs)
probs = torch.nn.functional.softmax(outputs.logits, dim=1)
print(probs)