加载中…
加载中…
ProsusAI/finbertcheckpoint-126/optimizer.pt 876 MBcheckpoint-168/optimizer.pt 876 MBcheckpoint-210/optimizer.pt 876 MBcheckpoint-42/optimizer.pt 876 MBcheckpoint-84/optimizer.pt 876 MBcheckpoint-126/model.safetensors 438 MBcheckpoint-168/model.safetensors 438 MBcheckpoint-210/model.safetensors 438 MBcheckpoint-42/model.safetensors 438 MBcheckpoint-84/model.safetensors 438 MBmodel.safetensors 438 MBcheckpoint-126/rng_state.pth 14.5 kBcheckpoint-168/rng_state.pth 14.5 kBcheckpoint-210/rng_state.pth 14.5 kBcheckpoint-42/rng_state.pth 14.5 kBcheckpoint-84/rng_state.pth 14.5 kBcheckpoint-126/training_args.bin 5.6 kBcheckpoint-168/training_args.bin 5.6 kBcheckpoint-210/training_args.bin 5.6 kBcheckpoint-42/training_args.bin 5.6 kBcheckpoint-84/training_args.bin 5.6 kBtraining_args.bin 5.6 kBcheckpoint-126/scheduler.pt 1.5 kBcheckpoint-168/scheduler.pt 1.5 kBcheckpoint-210/scheduler.pt 1.5 kBcheckpoint-42/scheduler.pt 1.5 kBcheckpoint-84/scheduler.pt 1.5 kBThis model is a fine-tuned version of ProsusAI/finbert on a custom dataset of ~500 financial news headlines written in Bahasa Indonesia. The task is 3-class sentiment classification: positive, neutral, and negative.
Try it here: https://huggingface.co/spaces/michaelmanurung/finbert-indonesia-demo
The base model is FinBERT, which itself is a BERT model pre-trained on financial texts. It has been fine-tuned using the Hugging Face transformers library with the following modifications:
The training dataset consists of 500 manually labeled financial news titles from Indonesian sources. Each entry is categorized as:
positive – bullish or growth-related headlinesneutral – factual or event-based reportingnegative – bearish or risk-indicative headlinesExample:
| Title | Label |
|---|---|
| IHSG diperkirakan rebound minggu ini | positive |
| BI umumkan suku bunga tetap | neutral |
| Rupiah melemah terhadap dolar AS | negative |
Evaluation was conducted on the intanm/indonesian-financial-sentiment-analysis benchmark dataset using accuracy, F1 score, precision, and recall metrics.
The benchmark compares:
michaelmanurung/finbert-indonesia (Indonesian financial sentiment model)ProsusAI/finbert (English financial sentiment model)ayameRushia/bert-base-indonesian-1.5G-sentiment-analysis-smsa (general Indonesian sentiment model)| Model | Accuracy | F1 Score | Precision | Recall |
|---|---|---|---|---|
| michaelmanurung/finbert-indonesia | 0.299 | 0.276 | 0.347 | 0.299 |
| ProsusAI/finbert | 0.285 | 0.134 | 0.560 | 0.285 |
| ayameRushia/bert-base-indonesian-1.5G-sentiment-analysis-smsa | 0.334 | 0.238 | 0.669 | 0.334 |
Metrics are reported using weighted averaging.
from transformers import pipeline
classifier = pipeline(
"text-classification",
model="michaelmanurung/finbert-indonesia",
tokenizer="michaelmanurung/finbert-indonesia"
)
result = classifier("IHSG turun tipis karena aksi ambil untung investor.")
print(result)
# Output: [{'label': 'LABEL_2', 'score': 0.89}] -> e.g. 'positive'