younsungsoo/klue-roberta-aiboms-v5
AiBoms (SafeInsight 본선 트랙) — KLUE-RoBERTa-base 기반 한국어 유해 메시지 분류 모델.
- 버전:
v5— v5 외부 강건 — 다양한 데이터 혼합 - 카테고리: violence / drug / self_harm / porn / safe (5-class)
- 베이스:
klue/roberta-base - 운영 백엔드:
backend/fastapi/app/services/klue_classifier.py(/v1/klue-classify,/v1/klue-compare,/v1/klue-health엔드포인트) - 앙상블 트리거: chat_room
__klue_ensemble__또는 환경변수KLUE_ENSEMBLE=1— Solar 1차 + KLUE 보조 분류 앙상블
사용
from transformers import AutoTokenizer, AutoModelForSequenceClassification
tok = AutoTokenizer.from_pretrained("younsungsoo/klue-roberta-aiboms-v5")
mdl = AutoModelForSequenceClassification.from_pretrained("younsungsoo/klue-roberta-aiboms-v5")
본선 시연 환경
운영 코드에서 환경변수 KLUE_MODELS_DIR 미설정 시 본 HF repo에서 자동 다운로드:
export KLUE_V4_REPO="younsungsoo/klue-roberta-aiboms-v5"
curl http://localhost:8000/v1/klue-health
# {"available": true, "model": "younsungsoo/klue-roberta-aiboms-v5", ...}
메타데이터
aiboms_meta.json에 학습 시점·하이퍼파라미터·평가 점수 기록.
라이선스
Apache 2.0. 베이스 모델 klue/roberta-base 정책 준수.
- Downloads last month
- 27