| # ๐ฎ KcELECTRA Steam Review Aspect Classifier (Phase 1) | |
| Fine-tuned model based on **beomi/KcELECTRA-base** | |
| for **Aspect-Based Sentiment Analysis (ABSA)** on Steam game reviews. | |
| ## ๐ Model Info | |
| - Base model: `beomi/KcELECTRA-base` | |
| - Task: Multi-label classification (6 aspects) | |
| - Labels: | |
| - STORY | |
| - OPTIMIZATION | |
| - GRAPHICS | |
| - PRICE_VALUE | |
| - BALANCE | |
| - ENGAGEMENT | |
| ## โ๏ธ Training | |
| - Dataset: Custom labeled Steam reviews (2,349 samples) | |
| - Loss: BCEWithLogitsLoss | |
| - Epochs: 5 | |
| - LR: 2e-5 | |
| - Batch size: 16 | |
| ## ๐ง Usage Example | |
| ```python | |
| from transformers import AutoTokenizer, AutoModelForSequenceClassification | |
| model = AutoModelForSequenceClassification.from_pretrained("Wing4/kcelectra-steam-aspect-classifier") | |
| tokenizer = AutoTokenizer.from_pretrained("Wing4/kcelectra-steam-aspect-classifier") | |
| inputs = tokenizer("๊ทธ๋ํฝ์ ์ข์ง๋ง ์ต์ ํ๊ฐ ๋ณ๋ก์ผ", return_tensors="pt") | |
| outputs = model(**inputs) | |
| print(outputs.logits.sigmoid()) # ๊ฐ ์ธก๋ฉด๋ณ ํ๋ฅ |