pruhtopia/multilingual-bert-toc-95k-dataset
Viewer • Updated • 94k • 9
How to use pruhtopia/bert-toc-classification-95k with Transformers:
# Use a pipeline as a high-level helper
from transformers import pipeline
pipe = pipeline("text-classification", model="pruhtopia/bert-toc-classification-95k") # Load model directly
from transformers import AutoTokenizer, AutoModelForSequenceClassification
tokenizer = AutoTokenizer.from_pretrained("pruhtopia/bert-toc-classification-95k")
model = AutoModelForSequenceClassification.from_pretrained("pruhtopia/bert-toc-classification-95k")loss: 0.5102838277816772
f1_macro: 0.605011586308457
f1_micro: 0.8910038281582305
f1_weighted: 0.8870714364293508
precision_macro: 0.6869883411452264
precision_micro: 0.8910038281582305
precision_weighted: 0.8858066104824025
recall_macro: 0.5550753643871188
recall_micro: 0.8910038281582305
recall_weighted: 0.8910038281582305
accuracy: 0.8910038281582305