🚀 微调版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。