FredZhang7
commited on
Commit
·
89f4c13
1
Parent(s):
3d00dcf
Update README.md
Browse files
README.md
CHANGED
@@ -2,10 +2,10 @@
|
|
2 |
license: creativeml-openrail-m
|
3 |
---
|
4 |
# Google Safesearch Mini Model Card
|
5 |
-
This model is trained
|
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 |
|
|
|
2 |
license: creativeml-openrail-m
|
3 |
---
|
4 |
# Google Safesearch Mini Model Card
|
5 |
+
This model is trained on 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's InceptionV3, the model achieves 94% accuracy on both train and test data.
|
9 |
|
10 |
<br>
|
11 |
|