Views
No views yet
indobenchmark/indobert-base-p2 untuk tugas klasifikasi sentimen review produk dalam Bahasa Indonesia dengan dua label (positif dan negatif). model di-pretrain terlebih dahulu menggunakan 78.536 baris ulasan produk agar lebih mengenali gaya bahasa informal yang umum digunakan dalam review, seperti singkatan "banget" → "bgt", "enggak" → "gk", dsb. kemudian model di finetune dengan Dataset yang terdiri dari 19.728 ulasan produk di e-commerce shopee dan juga berasal dari ulasan sintesis menggunakan LLM Gemini.POSITIF dan NEGATIF.TrainingArguments) yang digunakan saat pelatihan:1from transformers import TrainingArguments
2
3training_args = TrainingArguments(
4 output_dir="./results",
5 num_train_epochs=3,
6 per_device_train_batch_size=4,
7 gradient_accumulation_steps=2,
8 per_device_eval_batch_size=4,
9 weight_decay=0.05,
10 eval_strategy="epoch",
11 save_strategy="epoch",
12 seed=42,
13 load_best_model_at_end=True,
14 metric_for_best_model="f1",
15 logging_dir="./logs",
16 report_to="tensorboard",
17 logging_steps=100,
18 warmup_ratio=0.05,
19)| Epoch | Training Loss | Validation Loss | Accuracy | F1 Score |
|---|---|---|---|---|
| 1 | 0.2670 | 0.2374 | 0.9564 | 0.9564 |
| 2 | 0.4904 | 0.2951 | 0.9356 | 0.9356 |
| 3 | 0.3650 | 0.2176 | 0.9442 | 0.9441 |
1# Load model dan tokenizer
2from transformers import AutoTokenizer, AutoModelForSequenceClassification, pipeline
3import torch
4
5# Load dari Hugging Face Hub
6model = AutoModelForSequenceClassification.from_pretrained("siRendy/indobert-analisis-sentimen-review-produk-datathon2025")
7tokenizer = AutoTokenizer.from_pretrained("siRendy/indobert-analisis-sentimen-review-produk-datathon2025")
8
9# Fungsi prediksi
10def predict_sentiment(text):
11 classifier = pipeline(
12 "text-classification",
13 model=model,
14 tokenizer=tokenizer,
15 device=0 if torch.cuda.is_available() else -1
16 )
17
18 result = classifier(text)[0]
19 return {
20 "sentiment": str(result["label"]),
21 "confidence": round(result["score"], 4)
22 }
23
24# Contoh penggunaan
25text = "aih jelek nya ngawur iki opo seh, mo nipu lu ya"
26prediction = predict_sentiment(text)
27print(prediction)
28