Ateeqq commited on
Commit
ba2765c
·
verified ·
1 Parent(s): 7707aa5

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +4 -6
README.md CHANGED
@@ -36,16 +36,10 @@ This model is fine-tuned for **NSFW image classification**. It classifies conten
36
 
37
  ---
38
 
39
- ### 📌 Epoch Training Results
40
-
41
- ![Epoch Results](https://huggingface.co/Ateeqq/nsfw-image-detection/resolve/main/all-epochs-results.png)
42
-
43
  ### 📌 Confusion Matrix
44
 
45
  ![Metrics](https://huggingface.co/Ateeqq/nsfw-image-detection/resolve/main/final-epoch-results.png)
46
 
47
- ![Confusion Matrix](https://huggingface.co/Ateeqq/nsfw-image-detection/resolve/main/confusion_matrix.png)
48
-
49
  ---
50
 
51
  ### 🏷️ Categories
@@ -103,6 +97,10 @@ for i, score in enumerate(probs[0]):
103
  | 4 | 0.0109 | 0.0437 | 98.61% |
104
  | 5 ✅ | 0.0001 | 0.0389 | **99.02%** |
105
 
 
 
 
 
106
  - **Training runtime**: 1h 21m 40s
107
  - **Final Training Loss**: 0.0727
108
  - **Steps/sec**: 0.11 | **Samples/sec**: 6.99
 
36
 
37
  ---
38
 
 
 
 
 
39
  ### 📌 Confusion Matrix
40
 
41
  ![Metrics](https://huggingface.co/Ateeqq/nsfw-image-detection/resolve/main/final-epoch-results.png)
42
 
 
 
43
  ---
44
 
45
  ### 🏷️ Categories
 
97
  | 4 | 0.0109 | 0.0437 | 98.61% |
98
  | 5 ✅ | 0.0001 | 0.0389 | **99.02%** |
99
 
100
+ ### 📌 Epoch Training Results
101
+
102
+ ![Epoch Results](https://huggingface.co/Ateeqq/nsfw-image-detection/resolve/main/all-epochs-results.png)
103
+
104
  - **Training runtime**: 1h 21m 40s
105
  - **Final Training Loss**: 0.0727
106
  - **Steps/sec**: 0.11 | **Samples/sec**: 6.99