Views
No views yet
| Epoch | Train Loss | Train Accuracy | Eval Loss | Eval Accuracy | Training Time | Validation Time |
|---|---|---|---|---|---|---|
| 1 | 0.2471 | 88.15% | 0.2107 | 91.31% | 7:55 min | 10 sec |
| 2 | 0.1844 | 90.41% | 0.2107 | 92.39% | 7:50 min | 10 sec |
| 3 | 0.1502 | 91.66% | 0.2135 | 93.14% | 7:51 min | 9 sec |
| 4 | 0.1285 | 92.50% | 0.2192 | 93.69% | 7:50 min | 10 sec |
| 5 | 0.1101 | 93.13% | 0.2367 | 94.14% | 7:48 min | 9 sec |
1import tensorflow as tf
2from transformers import TFAutoModelForSequenceClassification, AutoTokenizer
3
4# Load model dan tokenizer
5model_name = "feverlash/Indonesian-SentimentAnalysis-Model" # Ganti dengan path model yang telah disimpan
6tokenizer = AutoTokenizer.from_pretrained(model_name)
7model = TFAutoModelForSequenceClassification.from_pretrained(model_name)
8
9# Fungsi untuk melakukan prediksi sentimen
10def predict(text):
11 sentiment_mapping = {
12 1: "positive",
13 0: "negative",
14 2: "neutral"
15 }
16
17 # Tokenisasi teks
18 inputs = tokenizer(
19 text,
20 return_tensors="tf",
21 truncation=True,
22 padding="max_length",
23 max_length=128
24 )
25
26 # Prediksi menggunakan model
27 outputs = model(inputs)
28 logits = outputs.logits
29
30 # Menghitung probabilitas
31 probabilities = tf.nn.softmax(logits).numpy()
32
33 # Menentukan label prediksi
34 predicted_index = int(tf.argmax(probabilities, axis=1).numpy()[0])
35 predicted_label = sentiment_mapping.get(predicted_index, "unknown")
36
37 # Keyakinan prediksi
38 confidence = probabilities[0][predicted_index]
39
40 print(f"Teks: {text}")
41 print(f"Prediksi label: {predicted_label} (Confidence: {confidence:.2f})")
42
43# Contoh penggunaan
44text = "aku sedang jalan-jalan di Yogyakarta"
45predict(text)