FredZhang7 commited on
Commit
70089b9
·
1 Parent(s): 6ec832e

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +1 -1
README.md CHANGED
@@ -5,7 +5,7 @@ license: creativeml-openrail-m
5
  This model is trained on InceptionV3 and 2,224,000 images scraped from Google Safesearch, Reddit, Imgur, and Github.
6
  It predicts the likelihood of an image being nsfw_gore, nsfw_suggestive, and safe.
7
 
8
- After 20 epochs on PyTorch, the model achieves 94% train and test accuracy.
9
 
10
  <br>
11
 
 
5
  This model is trained on InceptionV3 and 2,224,000 images scraped from Google Safesearch, Reddit, Imgur, and Github.
6
  It predicts the likelihood of an image being nsfw_gore, nsfw_suggestive, and safe.
7
 
8
+ After 20 epochs on PyTorch, the model achieves 94% accuracy on both train and test data.
9
 
10
  <br>
11