Indobert Emotion Classification

🚀 印尼語BERT情感分類模型

本項目提供了一個基於印尼語BERT（IndoBERT）的情感分類模型，可用於對印尼語文本進行情感分類。通過簡單的代碼調用，就能輕鬆實現文本情感分類功能。

🚀 快速開始

導入模型

以下代碼展示瞭如何導入該情感分類模型：

from transformers import BertForSequenceClassification, BertTokenizer, BertConfig

tokenizer = BertTokenizer.from_pretrained("thoriqfy/indobert-emotion-classification")
config = BertConfig.from_pretrained("thoriqfy/indobert-emotion-classification")
model = BertForSequenceClassification.from_pretrained("thoriqfy/indobert-emotion-classification", config=config)

使用模型

以下代碼展示瞭如何使用該模型進行文本情感分類：

from transformers import pipeline

nlp = pipeline("text-classification", model="thoriqfy/indobert-emotion-classification")

results = nlp("Your input text here")