README.md

---
license: mit
---
# Instagram Caption Classifier (Fine-tuned BERT)

Model ini merupakan hasil fine-tuning BERT (`bert-base-uncased`) untuk klasifikasi caption Instagram. Model ini dilatih menggunakan data caption yang telah dilabeli, sehingga dapat digunakan untuk mengklasifikasikan jenis caption pada postingan Instagram.

## Dataset
- Data: Caption Instagram berbahasa Indonesia, sudah dilabeli sesuai kategori relevan.
- Format: CSV, kolom utama: `post_caption`, `label_1`.
- Tersedia juga dataset komentar dan caption influencer Instagram yang sudah kami scrapping di Kaggle: [Kaggle Dataset - Instagram Influencer and Brand](https://www.kaggle.com/datasets/azrilfahmiardi/instagram-influencer-and-brand)

## Cara Penggunaan

```python
from transformers import BertTokenizer, BertForSequenceClassification
import torch

model = BertForSequenceClassification.from_pretrained("AzrilFahmiardi/instagram-caption-classifier-bert")
tokenizer = BertTokenizer.from_pretrained("AzrilFahmiardi/instagram-caption-classifier-bert")

text = "Caption contoh di sini"
inputs = tokenizer(text, return_tensors="pt", truncation=True, padding=True)
outputs = model(**inputs)
pred = torch.argmax(outputs.logits, dim=1)
```

## Label Kategori
Label yang digunakan dapat dilihat pada file `label_names.json` di repo ini.

## Lisensi
Model ini hanya untuk keperluan riset dan non-komersial.