Roberta Swahili News Classification

🚀 斯瓦希里語新聞分類RoBERTa模型

本模型藉助斯瓦希里語新聞數據，對預訓練的RoBERTa模型進行微調，用於斯瓦希里語新聞分類任務。模型在相關評估指標上表現良好，具有較高的準確性。

🚀 快速開始

此模型使用HuggingFace的Flax框架進行訓練，是由HuggingFace組織的JAX/Flax社區周的一部分。所有訓練工作均在由谷歌雲團隊贊助的TPUv3 - 8虛擬機上完成。

本任務以該模型為基礎進行微調。

💻 使用示例

基礎用法

from transformers import AutoTokenizer, AutoModelForSequenceClassification
  
tokenizer = AutoTokenizer.from_pretrained("flax-community/roberta-swahili-news-classification")

model = AutoModelForSequenceClassification.from_pretrained("flax-community/roberta-swahili-news-classification")

評估指標：

Eval metrics: {'accuracy': 0.9153416415986249}