Upload README.md with huggingface_hub

971dd58 verified 5 months ago

1.78 kB

language:
  - id
tags:
  - text-classification
  - query-routing
  - rag
  - indonesian
  - kepegawaian
  - asn
base_model: indobenchmark/indobert-lite-base-p2
pipeline_tag: text-classification
license: mit

iASN Query Router

Classifier ringan untuk merouting query pengguna ke RAG pipeline atau Agent pada sistem chatbot regulasi kepegawaian ASN.

Dikembangkan untuk sistem iASN — BKPSDM Kabupaten Sumenep, Jawa Timur.

Label

Label	Deskripsi
`RAG`	Pertanyaan regulasi, prosedur, aturan kepegawaian (dijawab dari dokumen)
`AGENT`	Query data personal pegawai — memerlukan akses ke SIMPEG/SIAGA
`CHITCHAT`	Salam, basa-basi, percakapan umum

Cara Pakai

from transformers import pipeline

classifier = pipeline(
    "text-classification",
    model="moh-affan/iasn-query-router",
    tokenizer="moh-affan/iasn-query-router",
)

result = classifier("Berapa sisa cuti saya tahun ini?")
# [{'label': 'AGENT', 'score': 0.97}]

Contoh Query

queries = [
    "Apa syarat kenaikan pangkat PNS?",       # → RAG
    "Berapa sisa cuti saya?",                  # → AGENT
    "Halo selamat pagi",                       # → CHITCHAT
    "Kapan SK mutasi saya terbit?",            # → AGENT
    "Bagaimana prosedur cuti melahirkan?",     # → RAG
]

for q in queries:
    r = classifier(q)
    print(f"{q:<45} → {r[0]['label']} ({r[0]['score']:.2%})")

Base Model

Fine-tuned dari indobenchmark/indobert-lite-base-p2

Tentang

Bagian dari penelitian tesis S2:

"Perancangan RAG-based Chatbot untuk Regulasi Kepegawaian ASN Kabupaten Sumenep"

Institut Sains dan Teknologi Terpadu Surabaya (ISTTS)