Получить массив целевых (y) значений в листе дерева регрессии - Scikit Learn

Я пытаюсь выполнить регрессию, используя ансамбли деревьев решений. Я обучил и вырастил множество массивов регрессии на массиве X_train,

С помощью X_testЯ понимаю, что я могу получить индекс листа, используя:

leaf_index_i_j = rf.estimators_[tree_j].apply(X_test[observation_i])

Теперь мне нужен вектор целевых значений y, которые принадлежат этому листу. Причина в том, что я хочу рассчитать распределение, а не среднее значение.

Пожалуйста помоги!

0 ответов

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