feverlash commited on
Commit
5c6c0c4
·
verified ·
1 Parent(s): c1d7e55

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +50 -1
README.md CHANGED
@@ -47,4 +47,53 @@ max_length = 128
47
 
48
  batch_size = 32
49
 
50
- eval_batch_size = 32
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
47
 
48
  batch_size = 32
49
 
50
+ eval_batch_size = 32
51
+
52
+ ## 🤖 How to use
53
+
54
+ ```python
55
+ import tensorflow as tf
56
+ from transformers import TFAutoModelForSequenceClassification, AutoTokenizer
57
+
58
+ # Load model dan tokenizer
59
+ model_name = "path/to/your/fine-tuned-model" # Ganti dengan path model yang telah disimpan
60
+ tokenizer = AutoTokenizer.from_pretrained(model_name)
61
+ model = TFAutoModelForSequenceClassification.from_pretrained(model_name)
62
+
63
+ # Fungsi untuk melakukan prediksi sentimen
64
+ def predict(text):
65
+ sentiment_mapping = {
66
+ 1: "positive",
67
+ 0: "negative",
68
+ 2: "neutral"
69
+ }
70
+
71
+ # Tokenisasi teks
72
+ inputs = tokenizer(
73
+ text,
74
+ return_tensors="tf",
75
+ truncation=True,
76
+ padding="max_length",
77
+ max_length=128
78
+ )
79
+
80
+ # Prediksi menggunakan model
81
+ outputs = model(**inputs)
82
+ logits = outputs.logits
83
+
84
+ # Menghitung probabilitas
85
+ probabilities = tf.nn.softmax(logits).numpy()
86
+
87
+ # Menentukan label prediksi
88
+ predicted_index = int(tf.argmax(probabilities, axis=1).numpy()[0])
89
+ predicted_label = sentiment_mapping.get(predicted_index, "unknown")
90
+
91
+ # Keyakinan prediksi
92
+ confidence = probabilities[0][predicted_index]
93
+
94
+ print(f"Teks: {text}")
95
+ print(f"Prediksi label: {predicted_label} (Confidence: {confidence:.2f})")
96
+
97
+ # Contoh penggunaan
98
+ text = "aku sedang jalan-jalan di Yogyakarta"
99
+ predict(text)