spark-nlp icon indicating copy to clipboard operation
spark-nlp copied to clipboard

MultiClassifierModel to return predicted labels ordered by highest to lowest probabilities

Open drbonis opened this issue 2 years ago • 2 comments

Is your feature request related to a problem? Please describe. No

Describe the solution you'd like A new parameter added to the Approach and Model like "setSortedByProb(true)" in case you want the model to return the labels ordered by descending predicted probability.

Additional context Could help in use cases where multiclass is desired but applying a "top n" approach instead of an arbitrary threshold.

drbonis avatar Nov 15 '21 11:11 drbonis

THanks @drbonis

I think sort by score and topN would be nice features

maziyarpanahi avatar Nov 25 '21 14:11 maziyarpanahi

This issue is stale because it has been open 120 days with no activity. Remove stale label or comment or this will be closed in 5 days

github-actions[bot] avatar Jun 17 '22 00:06 github-actions[bot]

This issue is stale because it has been open 120 days with no activity. Remove stale label or comment or this will be closed in 5 days

github-actions[bot] avatar Oct 20 '22 00:10 github-actions[bot]