Views
No views yet
beomi/KcELECTRA-base-v2022
학습에는 KMHAS 한국어 혐오 표현 데이터셋 사용beomi/KcELECTRA-base-v2022| Metric | Value |
|---|---|
| Accuracy | 0.91 |
| Precision | 0.91 |
| Recall | 0.91 |
| F1-score | 0.91 |
1from transformers import AutoTokenizer, AutoModelForSequenceClassification
2
3model = AutoModelForSequenceClassification.from_pretrained("now100/kmhas_electra_binary")
4tokenizer = AutoTokenizer.from_pretrained("now100/kmhas_electra_binary")
5
6text = "개새끼들이 나라를 망치고 있다."
7inputs = tokenizer(text, return_tensors="pt")
8outputs = model(**inputs)
9
10label = outputs.logits.argmax(dim=1).item()
11print("예측 결과:", "non-hate" if label == 1 else "hate")