OwenElliott commited on
Commit
6fe4bdc
1 Parent(s): 9215e4b

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +7 -0
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.