How to use kyLELEng/bert-ag-news-classifier with Transformers:
# Use a pipeline as a high-level helper from transformers import pipeline pipe = pipeline("text-classification", model="kyLELEng/bert-ag-news-classifier")
# Load model directly from transformers import AutoTokenizer, AutoModelForSequenceClassification tokenizer = AutoTokenizer.from_pretrained("kyLELEng/bert-ag-news-classifier") model = AutoModelForSequenceClassification.from_pretrained("kyLELEng/bert-ag-news-classifier")
d4610af ae2279b
1
2
3
4
5
6
,pred_World,pred_Sports,pred_Business,pred_Sci/Tech true_World,1814,8,44,34 true_Sports,11,1877,6,6 true_Business,37,8,1732,123 true_Sci/Tech,27,6,100,1767