wuriyanto's picture
Update README.md
a0b496e verified
---
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
```python
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}")
```