NSFW Text Classifier

🚀 微調版DistilRoBERTa-base用於不適宜內容分類

本項目基於DistilRoBERTa-base模型進行微調，旨在對不適宜工作場景（NSFW）的文本內容進行分類，能夠精準識別不恰當和不專業的文本，為內容審核等場景提供有力支持。

🚀 快速開始

模型描述

DistilBERT是一個用於情感分析的Transformer模型。本項目針對Reddit帖子對該模型進行了微調，以實現對不適宜工作場景（NSFW）內容的分類，特別是那些被認為不恰當和不專業的文本。模型可預測兩個類別，即不適宜工作場景（NSFW）和適宜工作場景（SFW）。

該模型是DistilBERT的微調版本，使用從（Reddit API）[https://praw.readthedocs.io/en/stable/]獲取的14317條Reddit帖子進行微調。

使用示例

基礎用法

from transformers import pipeline
classifier = pipeline("sentiment-analysis", model="michellejieli/NSFW_text_classification")
classifier("I see you’ve set aside this special time to humiliate yourself in public.")

Output:
[{'label': 'NSFW', 'score': 0.998853325843811}]

📞 聯繫我們

如果您有任何問題或反饋，請聯繫 michelle.li851@duke.edu。