Цель
Я работаю над многоцелевой регрессией, и я хотел бы сделать поиск по сетке (с функцией scikit learn grid search cv), чтобы найти лучшие параметры для моегомодель.Проблема в том, что для многоцелевой регрессии мне нужно оптимизировать оценку aRRMSE, которая принимает 3 аргумента: Y_pred, Y_train и Y_test.С помощью поиска по сетке мы можем поставить только оценку, которая принимает два аргумента (Y_pred и Y_test).
Вопрос
Знаете ли вы, как я могу сказать, что поиск по сеткемой третий параметр - Y_train?Или вы знаете способ сделать поиск по сетке с aRRMSE в качестве оценки?