Update README.md

README.md

@@ -2,8 +2,64 @@
 tags:
 - model_hub_mixin
 - pytorch_model_hub_mixin
+license: mit
+base_model:
+- microsoft/resnet-18
 ---
 
+# ResNetModelFT for Skin Cancer Classification
+
+## Model Details
+
+- **Model Architecture:** ResNet-18  
+- **Framework:** PyTorch  
+- **Input Shape:** 224x224 RGB images  
+- **Number of Parameters:** ~11.7M (ResNet-18 pretrained model)  
+- **Output:** Multi-class classification (9 classes)
+
+## Model Description
+
+This model uses **ResNet-18**, a well-known deep residual network, pre-trained on ImageNet. The model is fine-tuned by replacing the fully connected layer to accommodate multi-class classification for **skin cancer detection**. Only the fully connected layer is trainable, while the convolutional layers of the ResNet model are frozen to retain pretrained features.
+
+The final model performs multi-class classification with 9 output classes corresponding to different skin cancer types.
+
+## Training Details
+
+- **Optimizer:** Adam  
+- **Batch Size:** 64  
+- **Loss Function:** Cross-Entropy Loss  
+- **Number of Epochs:** 10  
+- **Dataset:** [Skin Cancer 9-Class Dataset](https://www.kaggle.com/datasets/nodoubttome/skin-cancer9-classesisic)
+
+### Metrics (Validation Set)
+
+| Class | Precision | Recall | F1-Score |
+|-------|-----------|--------|----------|
+| 0     | 1.00      | 0.06   | 0.12     |
+| 1     | 0.45      | 0.31   | 0.37     |
+| 2     | 0.57      | 0.25   | 0.35     |
+| 3     | 0.00      | 0.00   | 0.00     |
+| 4     | 0.32      | 1.00   | 0.48     |
+| 5     | 0.31      | 0.25   | 0.28     |
+| 6     | 0.50      | 0.67   | 0.57     |
+| 7     | 0.20      | 0.06   | 0.10     |
+| 8     | 0.14      | 1.00   | 0.24     |
+
+- **Overall Accuracy:** 0.31  
+- **Macro Average Precision:** 0.39  
+- **Macro Average Recall:** 0.40  
+- **Macro Average F1-Score:** 0.28  
+- **Weighted Average Precision:** 0.40  
+- **Weighted Average Recall:** 0.31  
+- **Weighted Average F1-Score:** 0.25  
+
+## License
+
+This model is released under the **MIT License**.
+
+---
+
+
 This model has been pushed to the Hub using the [PytorchModelHubMixin](https://huggingface.co/docs/huggingface_hub/package_reference/mixins#huggingface_hub.PyTorchModelHubMixin) integration:
 - Library: [More Information Needed]
 - Docs: [More Information Needed]