thai-column-classifier 0.1.0

Classify and detect sensitive columns in Thai datasets (CID, PDPA)

thai-column-classifier

A Python library for detecting and classifying sensitive columns in Thai datasets, designed to support PDPA (Personal Data Protection Act) compliance.

Features

• CID detection — identifies Thai national ID (เลขบัตรประชาชน) columns via exact match, fuzzy match, semantic similarity, and value pattern (13-digit checksum)
• Sensitive column detection — classifies columns as FULLNAME, PREFIX, FIRSTNAME, LASTNAME, EMAIL, ADDRESS_SHORT, or ADDRESS_FULL
• Pluggable LLM providers — Ollama (local), OpenAI, Claude, HuggingFace
• Pluggable semantic providers — local sentence-transformers or HuggingFace Inference API
• Privacy-first by default — all inference runs locally out of the box (Ollama + sentence-transformers); no data sent to external APIs

Detection pipeline

Each column goes through up to 4 stages (stops early if confident):

1. Exact match — keyword list
2. Fuzzy match — rapidfuzz ratio / partial_ratio / token_sort_ratio
3. Semantic — embedding cosine similarity
4. LLM — prompt-based classification as final fallback

Output decisions

Detector	Decision	Meaning
CID	auto_hash	Column is a Thai national ID — hash it
Sensitive	masking	Mask the entire value (****)
Sensitive	partial_masking	Mask street-level part only, keep district/province
Both	pass	Not sensitive

Supported input formats

The classifier operates on a pandas DataFrame — it is file-format agnostic. The load_file() helper in main.py supports:

Format	Extensions
CSV	.csv
Excel	.xlsx, .xls

To support additional formats (Parquet, JSON, database queries, etc.), you only need to extend load_file(). The classifier itself works unchanged as long as you pass it a DataFrame.

Installation

pip install thai-column-classifier

Install with optional providers:

pip install thai-column-classifier[ollama]     # Ollama (local LLM)
pip install thai-column-classifier[semantic]   # local sentence-transformers
pip install thai-column-classifier[openai]     # OpenAI
pip install thai-column-classifier[claude]     # Anthropic Claude
pip install thai-column-classifier[all]        # everything

Quick start

from thai_column_classifier import IDColumnClassifier, IDColumnInput
from thai_column_classifier import SensitiveColumnClassifier, SensitiveColumnInput, OllamaProvider

# CID detector
cid_clf = IDColumnClassifier()
result = cid_clf.classify(IDColumnInput(
    column_name="เลขบัตรประชาชน",
    sample_values=["1101700203451"]
))
print(result.decision)  # auto_hash

# Sensitive column detector — fully local by default (LocalSemanticProvider + OllamaProvider)
sensitive_clf = SensitiveColumnClassifier()
result = sensitive_clf.classify(SensitiveColumnInput(
    column_name="ชื่อ-นามสกุล",
    sample_values=["สมชาย ใจดี"]
))
print(result.decision)  # masking

Environment variables

HF_TOKEN=hf_...          # for HuggingFace providers
OPENAI_API_KEY=sk-...    # for OpenAI provider
ANTHROPIC_API_KEY=...    # for Claude provider

Running tests

python test_check_id.py
python test_check_sensitive.py
python main.py

Project structure

.
├── thai_column_classifier/
│   ├── __init__.py
│   ├── classifier.py                     # Unified ThaiColumnClassifier
│   ├── thai_id_column_detector.py        # CID classifier
│   └── thai_sensitive_column_detector.py # Sensitive column classifier
├── data/                                 # Test datasets
├── pyproject.toml
├── main.py
├── test_check_id.py
└── test_check_sensitive.py