Views
No views yet
| Model | Accuracy | Precision (Macro) | Recall (Macro) | F1 (Macro) | Type |
|---|---|---|---|---|---|
| IndoBERT | 0.9989 | 0.9989 | 0.9989 | 0.9989 | Transformer |
| Linear SVM | 0.9819 | 0.9820 | 0.9817 | 0.9818 | TF-IDF + SVM |
| Logistic Regression | 0.9782 | 0.9787 | 0.9777 | 0.9781 | TF-IDF + LR |
| Random Forest | 0.9765 | 0.9768 | 0.9760 | 0.9764 | TF-IDF + RF |
| Multinomial Naive Bayes | 0.9398 | 0.9414 | 0.9381 | 0.9393 | TF-IDF + NB |
1pip install gradio scikit-learn transformers torch
2python app.py1import pickle
2from transformers import AutoTokenizer, AutoModelForSequenceClassification
3import torch
4
5# For TF-IDF models (Logistic Regression, SVM, RF, NB)
6with open('logreg_model.pkl', 'rb') as f:
7 model = pickle.load(f)
8
9# Load vectorizer
10with open('tfidf_vectorizer.pkl', 'rb') as f:
11 vectorizer = pickle.load(f)
12
13text = "Your Indonesian news text here"
14X = vectorizer.transform([text])
15prediction = model.predict(X) # 0 for FAKTA, 1 for HOAX
16
17# For IndoBERT
18tokenizer = AutoTokenizer.from_pretrained('indobert_model')
19model = AutoModelForSequenceClassification.from_pretrained('indobert_model')
20
21inputs = tokenizer(text, return_tensors="pt", truncation=True, max_length=128)
22with torch.no_grad():
23 outputs = model(**inputs)
24 prediction = torch.argmax(outputs.logits, dim=-1).item()