ylecun/mnist
Viewer • Updated • 70k • 139k • 251
How to use MY555/MNIST_FINETUNED_BERT with Transformers:
# Use a pipeline as a high-level helper
from transformers import pipeline
pipe = pipeline("image-classification", model="MY555/MNIST_FINETUNED_BERT")
pipe("https://huggingface.co/datasets/huggingface/documentation-images/resolve/main/hub/parrots.png") # Load model directly
from transformers import AutoImageProcessor, AutoModelForImageClassification
processor = AutoImageProcessor.from_pretrained("MY555/MNIST_FINETUNED_BERT")
model = AutoModelForImageClassification.from_pretrained("MY555/MNIST_FINETUNED_BERT")loss: 0.05538161098957062
f1_macro: 0.9826821427792549
f1_micro: 0.983
f1_weighted: 0.9829770926906823
precision_macro: 0.9826130274580558
precision_micro: 0.983
precision_weighted: 0.9830119841075224
recall_macro: 0.9828091028515983
recall_micro: 0.983
recall_weighted: 0.983
accuracy: 0.983