redact-ner 0.6.8

Named Entity Recognition for PII detection using ONNX Runtime
Documentation

NER-based PII Recognition using ONNX Runtime

This crate provides Named Entity Recognition (NER) capabilities for PII detection using quantized ONNX models for efficient inference.

Features

  • ONNX Runtime integration for model inference
  • Support for quantized int8 models
  • Token-based NER with entity span detection
  • Compatible with various NER model architectures (BERT, RoBERTa, etc.)

Example

use redact_ner::NerRecognizer;
use redact_core::recognizers::Recognizer;

// Load model
let recognizer = NerRecognizer::from_file("model.onnx").unwrap();

// Analyze text
let text = "John Doe works at Acme Corp in New York";
let results = recognizer.analyze(text, "en").unwrap();

for result in results {
    println!("{:?}: {}", result.entity_type, result.text.unwrap());
}