| Входной файл: | Стандартный вход | Ограничение времени: | 10 сек | |
| Выходной файл: | Стандартный выход | Ограничение памяти: | 512 Мб |
Требуется реализовать следующие функцию на языке Python.
def tree_split(X, y, criterion) # col, row of best split
X — двумерный np.array — обучающая выборка
y — одномерный np.array — значения классов в обучающей выборке
criterion — строковое значение — вид критерия 'var', 'gini' или 'entropy'
tree_split должен возвращать номер признака и номер значения из обучающей выборки, которое будет использоваться в качестве порогового
Таким образом, tree_split возвращает наилучшее бинарное разделение по правилу вида xcol ≤ X[row, col]
Код должен содержать только реализацию функции.