Bert Toxic Comment Classification

🚀 毒性評論分類BERT模型

本項目是一個經過微調的文本分類模型，基於bert-base-uncased，專門用於識別毒性評論，能有效助力社區管理、內容審核等工作。

🚀 快速開始

模型描述

此模型是 bert-base-uncased 的微調版本，用於對毒性評論進行分類。

使用示例

基礎用法

from transformers import BertForSequenceClassification, BertTokenizer, TextClassificationPipeline

model_path = "JungleLee/bert-toxic-comment-classification"
tokenizer = BertTokenizer.from_pretrained(model_path)
model = BertForSequenceClassification.from_pretrained(model_path, num_labels=2)

pipeline = TextClassificationPipeline(model=model, tokenizer=tokenizer)
print(pipeline("You're a fucking nerd."))

訓練數據

訓練數據來自這個 Kaggle競賽。我們使用 train.csv 中90%的數據來訓練模型。

評估結果

該模型在1500行的保留測試集上達到了0.95的AUC值。

信息表格

許可證

本模型採用AFL-3.0許可證。