Advance-Ali commited on
Commit
f79ab5d
·
verified ·
1 Parent(s): 729f3c8

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +7 -4
README.md CHANGED
@@ -48,9 +48,9 @@ Classification Report:
48
 
49
  accuracy 0.80 3068
50
 
51
- macro avg 0.71 0.70 0.70 3068
52
-
53
- weighted avg 0.80 0.80 0.80 3068
54
 
55
  ## 🧑‍💻 How to Use
56
 
@@ -102,6 +102,9 @@ model.eval()
102
  ```
103
 
104
  ## ⚠️ Limitations and Considerations
 
105
  Input Dimensions: Make sure your input images are resized to the expected dimensions (100x100) before inference.
 
106
  Number of Classes: The trained model supports exactly 7 classes as defined in the training dataset.
107
- Output: The model output should be a probability of each of the 7 face type labels. Don't forget to use the softmax function to make predictions. Note that softmax is not used in the last layer of this model's architecture.
 
 
48
 
49
  accuracy 0.80 3068
50
 
51
+ macro avg 0.71 0.70 0.70 3068
52
+
53
+ weighted avg 0.80 0.80 0.80 3068
54
 
55
  ## 🧑‍💻 How to Use
56
 
 
102
  ```
103
 
104
  ## ⚠️ Limitations and Considerations
105
+
106
  Input Dimensions: Make sure your input images are resized to the expected dimensions (100x100) before inference.
107
+
108
  Number of Classes: The trained model supports exactly 7 classes as defined in the training dataset.
109
+
110
+ Output: The model output should be a probability of each of the 7 face type labels. Don't forget to use the softmax function to make predictions. Note that softmax is not used in the last layer of this model's architecture.