wuriyanto's picture
Update README.md
a0b496e verified
metadata
license: mit
language:
  - id
base_model:
  - sentence-transformers/paraphrase-multilingual-mpnet-base-v2
library_name: setfit
tags:
  - setfit
  - sentence-transformers
  - sentiment
  - indonesian

SetFit Indonesian Sentiment Analysis

Model Description

This model uses the sentence-transformers/paraphrase-multilingual-mpnet-base-v2 base model, which is trained with the Indonesian sentiment dataset.

In v1, this model can predict 2 sentiments. Namely baik and buruk sentiments.

Usage

from setfit import SetFitModel

model = SetFitModel.from_pretrained("wuriyanto/setfit-paraphrase-multilingual-mpnet-base-v2-indonesian-sentiment-v1")

texts_to_classify = ["Pemerintah kurang memperhatikan kesejahteraan masyarakat, sehingga banyak kriminal dimana-mana",
                     "Makanan yang dihidangkan cukup enak, beberapa orang juga menyampaikan jika fasilitas, terutama meja, kursi dan kamar mandi sangat bersih.",
                     "Tempat perbelanjaan tersebut secara harga cukup terjangkau, dan tidak jauh dari tempat tinggal mahasiswa",
                     "Suka dengan layanan yang diberikan. Harga juga cukup terjangkau bagi masyarakat menengah ke bawah.", 
                     "Dengan banyaknya barang import murah yang masuk ke tengah-tengah masyarakat, menyebabkan beberapa UMKM bangkrut"]
predictions = model.predict(texts_to_classify)

for text, prediction in zip(texts_to_classify, predictions):
    print(f"Text: {text} -> Sentiment: {prediction}")