How to obtain the vertex of multi-classification roc graph

salesforce / TransmogrifAI

TransmogrifAI (pronounced trăns-mŏgˈrə-fī) is an AutoML library for building modular, reusable, strongly typed machine learning workflows on Apache Spark with minimal hand-tuning

https://transmogrif.ai

BSD 3-Clause "New" or "Revised" License

2.24k stars 392 forks source link

How to obtain the vertex of multi-classification roc graph #454

Closed shenzgang closed 4 years ago

shenzgang commented 4 years ago

How to obtain the vertex of multi-classification roc graph？

tovbinm commented 4 years ago

TransmogrifAI exposes MultiClassificationMetrics computed by underlying MulticlassClassificationEvaluator in Apache Spark. Particular vertices are not available.

You can implement a custom evaluator either by using a handy factory Evaluators.MultiClassification.custom or by extending OpMultiClassificationEvaluatorBase