privacy-filter-tw — 部署 bundle
Fine-tuned openai/privacy-filter 用於台灣 PII 去識別化的完整部署包。
含權重 + Docker images + 執行腳本,可離線在 CPU 或 CUDA GPU 上跑。
受管控 Repo(Manual Review):申請存取後由 owner 手動核准。
內容
| 檔案 | 大小 | 說明 |
|---|---|---|
model/ |
2.6 GB | v3 fine-tuned checkpoint (OPF-native format) |
ntts_cpu_amd64.tar |
320 MB | Docker image (CPU-only, python:3.11-slim + torch cpu) |
ntts_cuda_amd64.tar |
4.0 GB | Docker image (pytorch/pytorch:2.7.0-cuda12.8-cudnn9) |
notes-scratch-2026/ |
工具 code:main.py, runtime.py, chunker.py, Dockerfiles | |
debs/ |
python3-venv + 7zip .deb (Ubuntu 22.04/24.04) 供離線裝 | |
RUN.sh |
一鍵執行腳本(自動偵測 CPU/GPU) | |
pii_categories.md |
19 類 PII 定義與準確度分佈 |
快速開始(Ubuntu 機器)
# 假設下載後放到 /home/ubuntu/pii_tool/
cd /home/ubuntu/pii_tool
# 載入 Docker image (擇一)
sudo docker load -i ntts_cuda_amd64.tar # GPU
# 或
sudo docker load -i ntts_cpu_amd64.tar # CPU
# 準備輸入資料在 /home/ubuntu/YOUR_INPUT_DIR/ (遞迴掃描 *.txt)
# 跑
sudo docker run --rm --gpus all \
-v /home/ubuntu/YOUR_INPUT_DIR:/data \
-v $(pwd)/model:/model \
ntts:cuda /data \
--model-dir /model \
--chunk-size 1000 \
--overlap 200
輸出:每個 <name>.txt → 同資料夾產生 <name>_masked.txt
CLI 參數
| Flag | Default | 說明 |
|---|---|---|
<input_dir> |
必填 | 遞迴掃描 *.txt |
--model-dir DIR |
(none) | 本地 model 路徑(離網必用) |
--chunk-size N |
1000 | Chunk 字元數 |
--overlap N |
200 | Chunk 重疊 |
--no-propagate |
off | 關閉全域 entity 傳播 |
--gpus auto|N|0 |
auto | 多卡並行;0 強制 CPU |
相關資源
- Base model:openai/privacy-filter
- Public 版 v3 weights:lianghsun/privacy-filter-tw
- Evaluation dataset:lianghsun/tw-PII-bench
- Training dataset:lianghsun/tw-PII-chat
License
Apache 2.0(繼承自 openai/privacy-filter)。