Halfotter/greensteel-frontend-material-classifier

GreenSteel Frontend Material Classifier

모델 설명

이 모델은 제철 공정에서 사용되는 원자재와 중간 제품을 분류하는 AI 모델입니다.

주요 기능

• 투입물명 수정 및 표준화
• 다국어 지원 (한국어, 영어)
• 정확한 매칭과 유사도 기반 예측

사용 예시

from transformers import AutoTokenizer, AutoModel
from app.ananke.model import XMLRoBERTaClassifier

# 모델 로드
model = XMLRoBERTaClassifier(model_dir="greensteel-frontend-material-classifier")

# 예측
result = model.predict("철ㄹ")
print(result)  # [{'label': '철', 'similarity': 100.0, 'match_type': 'exact_input'}]

학습 데이터

• 총 라벨 수: 14개
• 학습 텍스트 수: 14개

라벨 목록

['환원철', '소결', '석회석', '열유입', '용선', '모래', 'EAF 탄소 전극', '냉각수', '빌렛', '블룸', '코크스', '윤활제', '형강', '철']

라이선스

MIT License