Description
Provide a MulticlassClassificationEvaluator with weighted F1-score to tune multiclass classifiers using Pipeline API.
MLLib already provides a MulticlassMetrics functionality which can be wrapped around a MulticlassClassificationEvaluator to expose weighted F1-score as metric.
The functionality could be similar to scikit(http://scikit-learn.org/stable/modules/generated/sklearn.metrics.f1_score.html) in that we can support micro, macro and weighted versions of the F1-score (with weighted being default)
Attachments
Issue Links
- relates to
-
SPARK-9911
User guide for MulticlassClassificationEvaluator
-
- Resolved
-
- links to
