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

Question

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

对于spark ML中的MulticlassClassificationEvaluator（），是否可以通过每个类标签获得精确度/召回率？

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

Answer 1

看看class documentation这似乎不可能，使用内置方法。

虽然不完全是你想要的，你可以在weightedPrecision方法中使用weightedRecalland metricName。这至少会解决阶级失衡问题。