Как сохранить порог из функции H2O model_performance в Python?

Question

Я использую H2O в Python и создаю модель GBM для двоичной целевой переменной (1 против 0).Модель работает хорошо, и я вижу порог на выходе.Но я хочу сохранить порог в переменной (мы можем назвать его cut_point).Поэтому, когда я получаю новый набор данных, я могу использовать пороговое значение для определения 1 или 0. Кто-нибудь делал это раньше?

Answer 1

Вы можете использовать find_threshold_by_max_metric

model.find_threshold_by_max_metric('f1', train=True, valid=False, xval=False)