Views
No views yet
bert-base-uncased) untuk klasifikasi komentar Instagram. Model ini dilatih menggunakan data komentar yang telah dilabeli, sehingga dapat digunakan untuk mengklasifikasikan jenis komentar pada postingan Instagram.comment, predicted_label.1from transformers import BertTokenizer, BertForSequenceClassification
2import torch
3
4model = BertForSequenceClassification.from_pretrained("AzrilFahmiardi/instagram-comments-classifier-bert")
5tokenizer = BertTokenizer.from_pretrained("AzrilFahmiardi/instagram-comments-classifier-bert")
6
7text = "Komentar contoh di sini"
8inputs = tokenizer(text, return_tensors="pt", truncation=True, padding=True)
9outputs = model(**inputs)
10pred = torch.argmax(outputs.logits, dim=1)label_names.json di repo ini.