Model: eliasalbouzidi/distilbert-nsfw-text-classifier
Author: eliasalbouzidi
Pipeline: text-classification
Library: transformers
Tags: transformers, safetensors, distilbert, text-classification, Transformers,  PyTorch, safety, innapropriate, DiffGuard, en, dataset:eliasalbouzidi/NSFW-Safe-Dataset, arxiv:2412.00064, base_model:distilbert/distilbert-base-uncased, base_model:finetune:distilbert/distilbert-base-uncased, license:apache-2.0, model-index, autotrain_compatible, endpoints_compatible, region:us
Base models (finetune): id: distilbert/distilbert-base-uncased
Card metadata: widget: text: A family hiking in the mountains
example_title: Safe, text: A child playing with a puppy
example_title: Safe, text: A couple kissing passionately in bed
example_title: Nsfw, text: A woman naked
example_title: Nsfw, text: A man killing people
example_title: Nsfw, text: A mass shooting
example_title: Nsfw
base_model: distilbert-base-uncased
license: apache-2.0
language: en
metrics: f1, accuracy, precision, recall
pipeline_tag: text-classification
tags: Transformers,  PyTorch, safety, innapropriate, distilbert, DiffGuard
datasets: eliasalbouzidi/NSFW-Safe-Dataset
model-index: name: NSFW-Safe-Dataset
results: task: name: Text Classification
type: text-classification
dataset: name: NSFW-Safe-Dataset
type: .
metrics: name: F1
type: f1
value: 0.974
verified: no, name: Accuracy
type: accuracy
value: 0.98
verified: no
Likes: 23 | All-time downloads: 282022
Created: 2024-04-07 01:10:13