infinitered/nsfwjs

Create a voting classifier ensemble for more advanced accuracy

Open

#491 opened on Mar 30, 2021

View on GitHub
 (1 comment) (0 reactions) (0 assignees)JavaScript (7,090 stars) (507 forks)batch import
enhancementhelp wanted

Description

Currently, we have two models. The more accurate one that has type 2 errors, and the slightly less accurate one that has far more type 1 errors

We can combine them with a voting classifier and get higher accuracy.

https://medium.com/@sanchitamangale12/voting-classifier-1be10db6d7a5

Thoughts?

Contributor guide

Tech stack
javascripttensorflow
Domain
machine learning
Issue type
feature
DifficultyEstimated implementation difficulty for a new contributor, from 1 for very small changes to 5 for expert-level work.
3
Estimated timeA rough time range for an experienced contributor to investigate, implement, test, and prepare a pull request.
1-2 days
Activity statusHow available the issue appears right now: fresh, active, stale, blocked, or waiting on maintainer input.
stale
ClarityHow clearly the issue explains the expected change, acceptance criteria, and next step.
needs investigation
Prerequisites
Understanding of existing NSFW models in the repoBasic knowledge of ensemble methods
Newbie friendlinessA 1-100 score estimating how approachable this issue is for first-time contributors.
25
Research direction
Examine the existing models in the repository (likely in /src or /models). Determine how to implement a voting classifier that aggregates predictions from both models. Consider using TensorFlow.js's built in methods or custom logic. Test the ensemble on a dataset to measure accuracy improvement. The issue does not specify implementation details, so further clarification from maintainers may be needed.
Create a voting classifier ensemble for more advanced accuracy · infinitered/nsfwjs#491 | Good First Issue