Views
No views yet
1>>>pip install torch
2>>>pip install transformers
3>>>from transformers import BertTokenizer, BertForSequenceClassification
4>>>import torch
5
6model_name = "radityapranata/absabert-keluhanpln-v3"
7tokenizer = BertTokenizer.from_pretrained(model_name)
8model = BertForSequenceClassification.from_pretrained(model_name)
9absa_pipeline = pipeline("sentiment-analysis", model=model, tokenizer=tokenizer)
10
11text = "informasi tambah daya" #"Token listrik tidak dapat terisi, kwh meter tulisan periksa."
12inputs = tokenizer(text, return_tensors="pt")
13result = absa_pipeline(text)
14
15outputs = model(**inputs)
16logits = outputs.logits
17
18get_aspect = torch.argmax(logits, dim=1).item()
19aspects = ["Layanan Pelanggan", "Ketersediaan Produk", "Kebijakan Usaha", "Pemulihan Layanan"]
20aspect = aspects[get_aspect]
21
22for aspect_result in result:
23 Sentiment = aspect_result["label"]
24 Score = aspect_result["score"]
25 print("Aspect:", aspect)
26 print(f"Sentiment: {Sentiment}, Score: {Score}")