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 許可證。