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)
- Downloads last month
- 24
Model tree for moh-affan/iasn-query-router
Base model
indobenchmark/indobert-lite-base-p2