GridSearchCV на собственной метрике для бинарной классификации

@Лебедева Саша · Регистрация: 17.11.2018

Классы 0 и 1. У каждой 1 своя заданная польза от 1 до 10 - power. Известна заранее

метрика для GridsearchCV считает сумму соответствующих значение из столбца power для верно предсказанного класса(1)

Python

1 2	def tot(y_true, y_pred, power): return power[(y_pred==1) & (y_true==1)].sum()

Такой вызов API:

Python

1 2	XGBgs = GridSearchCV(XGB, param_grid, scoring = make_scorer(tot, greater_is_better=True)) XGBgs.fit(x, y, power)

вызывает ошибку

Как правильно обучить GridSearchCV на своей метрике с power?

@Катафалк · 03.12.2020, 18:44

Как написано в документации https://scikit-learn.org/stabl... corer.html
ваш power нужно писать в make_scorer, а не в fit.

@Катафалк 107 / 217 / 10 Регистрация: 21.06.2019 Сообщений: 1,121
	03.12.2020, 18:44	2
	Как написано в документации https://scikit-learn.org/stabl... corer.html ваш power нужно писать в make_scorer, а не в fit. 0

Опции темы