NSFW Text Classifier

🚀 ファインチューニングされたDistilRoBERTa-baseによる不適切コンテンツ分類

このモデルは、センチメント分析を行うTransformerモデルであるDistilBERTを、Redditの投稿データでファインチューニングしたものです。主に、不適切な内容や非専門的なテキストを含む不適切コンテンツ（NSFW）を分類する目的で開発されています。モデルは、不適切コンテンツ（NSFW）または適切コンテンツ（SFW）の2つのクラスを予測します。

🚀 クイックスタート

このモデルを使用するには、以下の手順に従ってください。

💻 使用例

基本的な使用法

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}]

📚 ドキュメント

モデルの詳細

このモデルは、DistilBERT のファインチューニング版です。

訓練データ

このモデルは、(Reddit API) [https://praw.readthedocs.io/en/stable/] から取得した14317件のReddit投稿でファインチューニングされています。

連絡先

質問やフィードバックがある場合は、michelle.li851@duke.edu までご連絡ください。