Nestech/electronic-base

Pre-trained model and tokenizer

from this repo mdhugol/indonesia-bert-sentiment-classification

Dataset

from this organization Dataset Electronic

Evaluation

How to Use

from transformers import pipeline
from transformers import AutoTokenizer, AutoModelForSequenceClassification

checkpoint = "Nestech/electronic-base"

model = AutoModelForSequenceClassification.from_pretrained(checkpoint)
tokenizer = AutoTokenizer.from_pretrained(checkpoint)

sentiment_analysis = pipeline("sentiment-analysis", model=model, tokenizer=tokenizer)


pos_text = "Sangat bahagia hari ini"
neg_text = "Dasar anak sialan!! Kurang ajar!!"
net_text = "aku dan kamu"

result_pos = sentiment_analysis(pos_text)
print(result_pos)

result_neg = sentiment_analysis(neg_text)
print(result_neg)

result_net = sentiment_analysis(net_text)
print(result_net)