spam-indobert

Spam Review Classifier (IndoBERT + Flask)

Sebuah proyek klasifikasi ulasan spam menggunakan model IndoBERT dan API Flask. Cocok untuk menyaring ulasan negatif atau spam secara otomatis.

---

Instalasi

1. Clone repo (opsional):

   git clone https://github.com/kamu/spam-review-classifier.git
   cd spam-review-classifier
   
   

2. Environment dan Instalasi dependensi:

   python -m venv venv
   source venv/bin/activate  # Windows: venv\Scripts\activate
   pip install -r requirements.txt
   

Training Model

Pastikan file datasetspam.csv berada di dalam folder data/ dengan struktur:

Lalu jalankan proses training:

python src/trainer.py

Hasil training akan disimpan di direktori:

model/filter_review_model/

Menjalankan API Flask

Setelah training selesai, jalankan API Flask dengan perintah:

python app/main.py

API akan aktif di alamat:

http://localhost:5000

Endpoint API

POST /filter-review

1. Request JSON:

   {
       "text": "Dapatkan pulsa gratis klik di sini"
   }
   

2. Response JSON:

   {
       "prediction": 1,
       "confidence": 0.9451,
       "processed_text": "dapatkan pulsa gratis klik di sini"
   }
   
   

prediction: 0 = bukan spam, 1 = spam confidence: probabilitas prediksi processed_text: teks yang sudah diproses

Catatan

Model menggunakan: indobenchmark/indobert-base-p2 Konfigurasi model dan training dapat diatur di src/config.py