Alirezamp/news-category-langchain
Viewer • Updated • 105k • 22 • 1
How to use Alirezamp/autotrain-albert-news-category-clf with Transformers:
# Use a pipeline as a high-level helper
from transformers import pipeline
pipe = pipeline("text-classification", model="Alirezamp/autotrain-albert-news-category-clf") # Load model directly
from transformers import AutoTokenizer, AutoModelForSequenceClassification
tokenizer = AutoTokenizer.from_pretrained("Alirezamp/autotrain-albert-news-category-clf")
model = AutoModelForSequenceClassification.from_pretrained("Alirezamp/autotrain-albert-news-category-clf", device_map="auto")loss: 0.303434818983078
f1_macro: 0.8907819659105525
f1_micro: 0.9081382385730212
f1_weighted: 0.908187488481894
precision_macro: 0.8875819460768244
precision_micro: 0.9081382385730212
precision_weighted: 0.9098266246359188
recall_macro: 0.8969158254453838
recall_micro: 0.9081382385730212
recall_weighted: 0.9081382385730212
accuracy: 0.9081382385730212
Base model
albert/albert-base-v2