Spark ML - MulticlassClassificationEvaluator - 我们可以通过每个类标签获得精确/召回吗?

Sam*_*Sam 10 machine-learning bigdata apache-spark apache-spark-ml multiclass-classification

我在Spark ML中使用随机森林进行多类预测.

对于spark ML中的MulticlassClassificationEvaluator(),是否可以通过每个类标签获得精确度/召回率?

目前,我只看到所有班级的精确/召回相结合.

Owl*_*ght 0

查看类文档,使用内置方法似乎不可能。

尽管不完全是您正在寻找的内容,但您可以在该方法中使用weightedPrecisionand 。这至少可以解决阶级不平衡的问题。weightedRecallmetricName