make_scorer в перекрестной проверке с многомерным выводом из классификатора

Я пытаюсь выполнить перекрестную проверку с помощью настраиваемой функции подсчета очков. Сложность в том, что мне нужен многомерный вывод моего классификатора в моей scorer_function как y_pred. В документации make_scorer сказано, что только двоичные массивы 1d принимаются как y_pred.

https://scikit-learn.org/stable/modules/generated/sklearn.metrics.make_scorer.html

Есть ли другой способ использовать многомерные выходные данные классификатора для моей функции счетчика при перекрестной проверке?

      def own_function(y, y_pred):
   score = "Do something with multi dimensional y_pred"
   return score

my_scorer = make_scorer(own_function, greater_is_better=False,needs_proba=True)

scores = cross_validate(model, 
                        X, 
                        y, 
                        scoring=my_scorer, 
                        cv=cv, 
                        n_jobs=-1,
                        return_estimator=True)

0 ответов

Другие вопросы по тегам