Skip to content

Releases: samuel-s-marques/nsfw-underage-filter

v1.0.0

21 Jul 14:28
Compare
Choose a tag to compare

This is an ONNX model for classifying Tavern V1/V2 character cards as potentially underage or adult. Trained on 374 adult characters and 295 underage characters, both from chub.ai.

Performance:

ROC-AUC Score: 0.97
Accuracy: 0.90 (89% rounded up)
Precision, Recall, and F1-score for both classes:

Adult:

  • Precision: 0.84
  • Recall: 0.96
  • F1-score: 0.90

Underage:

  • Precision: 0.95
  • Recall: 0.82
  • F1-score: 0.88

Weighted average:

  • Precision: 0.89
  • Recall: 0.89
  • F1-score: 0.89