Upload README.md with huggingface_hub

README.md

@@ -1,3 +1,4 @@
+
 ---
 language: en
 license: mit
@@ -5,26 +6,27 @@ datasets: [Cats-Dogs]
 metrics: [accuracy, f1, precision, recall]
 ---
 
-# Transfer Learning On Alexnet:
+# 🐱🐶 Transfer Learning on AlexNet for Cats vs. Dogs Classification
 
-This model uses Transfer Learning for a 2 labelled class Cats-Dogs classification.
+This model fine-tunes **AlexNet** using Transfer Learning to classify images into two categories: **Cats** and **Dogs**.
 
-- Pre-trained model used:  Alexnet.
-- Dataset used:            Cats_Dogs.
-- batch_size = 8
-- learning_rate = 0.001
-- epochs = 5
+## **📝 Model Details**
+- **Pre-trained Model:** AlexNet  
+- **Dataset Used:** Cats-Dogs  
+- **Batch Size:** 8  
+- **Learning Rate:** 0.001  
+- **Epochs:** 5  
 
-## **Evaluation Results**
+---
 
-### 📌 **Before Transfer Learning**
+## **📌 Baseline Performance (Before Transfer Learning)**  
+**Validation Accuracy:** **40.66%**  
+- **Precision:** 0.3983  
+- **Recall:** 0.4066  
+- **F1-score:** 0.3942  
 
-** Accuracy:** 40.66%
-- Precision: 0.3983, 
-- Recall: 0.4066, 
-- F1-score: 0.3942,
+**Confusion Matrix:**
 
-- Confusion Matrix:
 [[ 659 1841]
  [1126 1374]]
 ---
@@ -40,7 +42,7 @@ This model uses Transfer Learning for a 2 labelled class Cats-Dogs classificatio
  [1006 8994]]
 
 
-** Validatio Accuracy:** 94.10%
+** Validation Accuracy:** 94.10%
 - Precision: 0.9425, 
 - Recall: 0.9410, 
 - F1-score: 0.9410,