diff --git "a/README.md" "b/README.md" --- "a/README.md" +++ "b/README.md" @@ -10,11 +10,11 @@ tags: - text-classification - generated_from_setfit_trainer widget: -- text: 주방 / 생활 / 이미용가전 > 커피머신 / 커피용품 > 에스프레소머신 > 업소용 -- text: 뷰티 > 스킨케어 > 클렌징 > 클렌징오일 -- text: 가전디지털 > 이미용가전 > 피부관리기기 > 여드름 / 피지관리 > 여드름치료기 -- text: 전동공구 > 전동드라이버 > seyoung -- text: 등산 > 아웃도어 > 배낭 > 모자 > 용품 > 반다나 > 스카프 > 마스크 +- text: 노트북 > msi > 블루라이트차단 +- text: 해외직구 > 건강식품 > 칼슘 +- text: 출산 / 육아용품 > 침구 / 수면용품 > 이불 / 담요 +- text: 생활가전 > 청소기 > 핸디청소기 +- text: 생활 > 건강 / 안마용품 > 온열 / 찜질용품 > 냉온주머니 / 핫팩 inference: true model-index: - name: SetFit with klue/roberta-base @@ -28,7 +28,7 @@ model-index: split: test metrics: - type: metric - value: 0.970703125 + value: 0.9797794117647058 name: Metric --- @@ -62,31 +62,31 @@ The model has been trained using an efficient few-shot learning technique that i ### Model Labels | Label | Examples | |:------|:---------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------| -| 8 | | -| 4 | | -| 11 | | -| 5 | | -| 14 | | -| 6 | | -| 15 | | -| 9 | | -| 17 | | -| 12 | | -| 10 | | -| 7 | | -| 16 | | -| 0 | | -| 3 | | -| 1 | | -| 13 | | -| 2 | | +| 10 | | +| 7 | | +| 4 | | +| 3 | | +| 11 | | +| 12 | | +| 8 | | +| 5 | | +| 6 | | +| 15 | | +| 16 | | +| 1 | | +| 14 | | +| 2 | | +| 17 | | +| 13 | | +| 9 | | +| 0 | | ## Evaluation ### Metrics | Label | Metric | |:--------|:-------| -| **all** | 0.9707 | +| **all** | 0.9798 | ## Uses @@ -106,7 +106,7 @@ from setfit import SetFitModel # Download from the 🤗 Hub model = SetFitModel.from_pretrained("setfit_model_id") # Run inference -preds = model("뷰티 > 스킨케어 > 클렌징 > 클렌징오일") +preds = model("해외직구 > 건강식품 > 칼슘") ```