pii-ja-ner-onnx

PII-JA-Detection (ONNX)

日本語PII（個人識別情報）検出モデルのONNXエクスポート版です。

モデル概要

• ベースモデル: tohoku-nlp/bert-base-japanese
• タスク: トークン分類 (BIO tagging)
• 対応PIIタイプ: 18種類

検出可能なPIIタイプ

使い方

import numpy as np
import onnxruntime as ort
from transformers import AutoTokenizer

# セッション作成
session = ort.InferenceSession("model.onnx")
tokenizer = AutoTokenizer.from_pretrained(".")

# 推論
text = "田中太郎の電話番号は090-1234-5678です。"
inputs = tokenizer(text, return_tensors="np", padding="max_length", max_length=512)
logits = session.run(None, {
    "input_ids": inputs["input_ids"],
    "attention_mask": inputs["attention_mask"],
})[0]

predictions = np.argmax(logits, axis=-1)

ラベルマッピング

BIOタグ方式（37ラベル）: O, B-PERSON_NAME, I-PERSON_NAME, ...

入出力仕様