Update README.md
Browse files
README.md
CHANGED
@@ -10,4 +10,28 @@ tags:
|
|
10 |
- sentence-transformers
|
11 |
- sentiment
|
12 |
- indonesian
|
13 |
-
---
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
10 |
- sentence-transformers
|
11 |
- sentiment
|
12 |
- indonesian
|
13 |
+
---
|
14 |
+
# SetFit Indonesian Sentiment Analysis
|
15 |
+
|
16 |
+
### Model Description
|
17 |
+
|
18 |
+
This model uses the `sentence-transformers/paraphrase-multilingual-mpnet-base-v2` base model, which is trained with the Indonesian sentiment dataset.
|
19 |
+
|
20 |
+
In v1, this model can predict 2 sentiments. Namely 'baik' and 'buruk' sentiments.
|
21 |
+
|
22 |
+
### Usage
|
23 |
+
```python
|
24 |
+
from setfit import SetFitModel
|
25 |
+
|
26 |
+
model = SetFitModel.from_pretrained("wuriyanto/setfit-paraphrase-multilingual-mpnet-base-v2-indonesian-sentiment-v1")
|
27 |
+
|
28 |
+
texts_to_classify = ["Pemerintah kurang memperhatikan kesejahteraan masyarakat, sehingga banyak kriminal dimana-mana",
|
29 |
+
"Makanan yang dihidangkan cukup enak, beberapa orang juga menyampaikan jika fasilitas, terutama meja, kursi dan kamar mandi sangat bersih.",
|
30 |
+
"Tempat perbelanjaan tersebut secara harga cukup terjangkau, dan tidak jauh dari tempat tinggal mahasiswa",
|
31 |
+
"Suka dengan layanan yang diberikan. Harga juga cukup terjangkau bagi masyarakat menengah ke bawah.",
|
32 |
+
"Dengan banyaknya barang import murah yang masuk ke tengah-tengah masyarakat, menyebabkan beberapa UMKM bangkrut"]
|
33 |
+
predictions = model.predict(texts_to_classify)
|
34 |
+
|
35 |
+
for text, prediction in zip(texts_to_classify, predictions):
|
36 |
+
print(f"Text: {text} -> Sentiment: {prediction}")
|
37 |
+
```
|