Japanese Sexual Moderation V2

🚀 日语性内容审核模型v2

本项目基于 studio-ousia/luke-japanese-large-lite 进行微调，构建了一个能够对短文进行性内容评分的模型。该模型通过回归学习，输出的分数大致在 0 - 1 之间，但也可能出现负值或超过 1 的情况。模型未对长文进行训练，输入时建议按换行符分割文本。

🚀 快速开始

模型评分区间说明

• 0.0 - 0.2：完全不涉及性内容
• 0.2 - 0.4：几乎不包含性内容
• 0.4 - 0.6：可能包含性内容
• 0.6 - 0.8：包含性内容
• 0.8 - 1.0：具有非常强的性暗示内容

💻 使用示例

基础用法

import torch
from transformers import AutoModelForSequenceClassification, AutoTokenizer

model_id = "oshizo/japanese-sexual-moderation-v2"

tokenizer = AutoTokenizer.from_pretrained(model_id)
model = AutoModelForSequenceClassification.from_pretrained(
    model_id,
    problem_type="regression"
)

texts = [
    "富士山は日本で一番高い山です。",
    "こんなになるまで我慢してたんだ...大変だったね",
    "俺は倒れる京子に近づき、担ぎ上げようと太ももの下に腕を通す。",
]

with torch.no_grad():
    encoding = tokenizer(texts, padding=True, max_length=64, return_tensors="pt")
    scores = model(**encoding).logits

# tensor([[-0.0045],
#         [ 0.2954],
#         [ 0.4391]])

📄 许可证

本项目采用 MIT 许可证。