Je fais une prédiction multiclasse avec random forest dans Spark ML.
Pour cet évaluateur de classification multiclasse () dans Spark ML, est-il possible d'obtenir la précision / le rappel pour chaque étiquette de classe?
Actuellement, je ne vois que la précision / le rappel combinés pour l'ensemble des classes.
En regardant la documentation de la classe (https://spark.apache.org/docs/1.6.2/api/java/org/apache/spark/ml/evaluation/MulticlassClassificationEvaluator.html), cela ne semble pas être possible en utilisant les méthodes intégrées.
Bien que ce ne soit pas exactement ce que vous recherchez, vous pouvez utiliser weightedPrecision et weightedRecall dans la méthode metricName. Cela permettra au moins de tenir compte des déséquilibres de classe.
Prograide est une communauté de développeurs qui cherche à élargir la connaissance de la programmation au-delà de l'anglais.
Pour cela nous avons les plus grands doutes résolus en français et vous pouvez aussi poser vos propres questions ou résoudre celles des autres.
