Downloads · 30 days
4
7% of all-time downloads
xyxing/dermai-efficientnet-b0
dermai-efficientnet-b0 is a machine learning model from xyxing. Use it for the machine learning task on the model card, and read the license before you ship it in a product.
Fine-tuned google/efficientnet-b0 for 7-class skin lesion classification on HAM10000. Part of the DermAI explainability project comparing CNN and Vision Transformer explanations.
Downloads · 30 days
4
7% of all-time downloads
All-time downloads
61
Public
Parameters
4.1M
16.3 MB on disk
Likes
0
Public
Click a slice to open those files.
.safetensors16.3 MB · 100%
From the Hugging Face model README
Fine-tuned google/efficientnet-b0 for 7-class skin lesion classification on HAM10000. Part of the DermAI explainability project comparing CNN and Vision Transformer explanations.
Hyperparameter tuned: phase1 lr = 4e-4, phase2 lr = 1e-4, weight decay = 0.01, batch size = 32, warmup ratio = 0.1
Best validation Macro F1 = 0.7073