OwenElliott
commited on
Commit
•
6fe4bdc
1
Parent(s):
9215e4b
Update README.md
Browse files
README.md
CHANGED
@@ -47,6 +47,13 @@ This model outperforms existing NSFW detectors on our dataset, here we provide a
|
|
47 |
|
48 |
![Evaluation against other models](./images/Comparison.png)
|
49 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
50 |
## Training Details
|
51 |
|
52 |
This model is a finetune of the [timm/vit_tiny_patch16_384.augreg_in21k_ft_in1k](https://huggingface.co/timm/vit_tiny_patch16_384.augreg_in21k_ft_in1k) model.
|
|
|
47 |
|
48 |
![Evaluation against other models](./images/Comparison.png)
|
49 |
|
50 |
+
### Thresholds and Precision vs Recall
|
51 |
+
|
52 |
+
Adjusting the threshold for the NSFW probability can let you trade off precision, recall, and accuracy. This maybe be useful in different applications where different degrees of confidence are required.
|
53 |
+
|
54 |
+
![Thresholded Evaluation](./images/ThresholdEvals.png)
|
55 |
+
![Precision and Recall Curves](./images/PrecisionRecallCurves.png)
|
56 |
+
|
57 |
## Training Details
|
58 |
|
59 |
This model is a finetune of the [timm/vit_tiny_patch16_384.augreg_in21k_ft_in1k](https://huggingface.co/timm/vit_tiny_patch16_384.augreg_in21k_ft_in1k) model.
|