Update README.md

README.md

@@ -10,15 +10,24 @@ tags:
 - ML
 - Ai
 ---
-Model Card: Facial Recognition Model
+Facial Recognition Model
+
 Model Information
+
 Name: Facial Means
+
 Type: Convolutional Neural Network (CNN)
+
 Framework: TensorFlow
+
 Dataset: Celebrity Faces Dataset
+
 Dataset Path: /content/drive/MyDrive/beard_dataset/celb_dataset/
+
 Model Save Path: /content/drive/MyDrive/beard_dataset/celebrity_model.h5
+
 Image Dimensions: 224 x 224 pixels
+
 Batch Size: 32
 Data Augmentation
 Rescale: 1./255
@@ -27,7 +36,8 @@ Zoom Range: 0.2
 Horizontal Flip: True
 Validation Split: 20%
 Model Architecture
-Layer (type)                 Output Shape              Param #
+Layer (type)
+Output Shape              Param #
 ===============================================================
 conv2d (Conv2D)               (None, 222, 222, 32)      896
 max_pooling2d (MaxPooling2D)  (None, 111, 111, 32)      0
@@ -39,6 +49,7 @@ flatten (Flatten)             (None, 86528)             0
 dense (Dense)                 (None, 128)               11075712
 dense_1 (Dense)               (None, 6)                 774
 ===============================================================
+
 Total params: 11,170,734
 Trainable params: 11,170,734
 Non-trainable params: 0
@@ -46,8 +57,11 @@ Model Compilation
 Optimizer: Adam
 Loss Function: Categorical Crossentropy
 Metrics: Accuracy
+
 Training
+
 Epochs: 10
+
 Steps per Epoch: Calculated based on the training dataset size and batch size.
 Validation Steps: Calculated based on the validation dataset size and batch size.
 Model Save