Как построить кривую ROC для мультикласса Xgboost с использованием Python? - PullRequest
0 голосов
/ 29 июня 2018

В моей модели классы [0,1,2,3]. Поскольку ROC используется в бинарной классификации или задачах мультикласса OneVsRest, я хочу построить кривую ROC для классов [0,1], [0,2], [0,3]. Что мне делать?

1 Ответ

0 голосов
/ 08 августа 2018

ROC-кривые смоделированы для бинарных задач. OneVsAll - это один из способов сделать это, когда ваш основной класс рассматривается как положительный ярлык, а другие как отрицательный. В соответствии с вашей проблемой вы хотите смоделировать проблему как OneVsOne, что хорошо. Поскольку это еще один способ заставить двоичные классификаторы работать для вашей мультиклассовой классификации. Внимание: метод OneVsOne вычислительно дорог. См. здесь для дальнейшего чтения.

Для классов [0,1,2] он будет возвращать [0,1], [0,2], [1,2], т.е. для 3 классов он будет возвращать 3 (3-1) / 2, то есть 3 классификатора , Теперь у вас есть 3 двоичных классификатора. Я надеюсь, что вам будет легко следить за тем, как получить кривые ROC с этой точки.

...