Как рассчитать доверительный интервал в h2o4gpu XGBClassifier?

Question

С GPU h2o я получаю следующий вывод из прогноза:

model.predict(data)

False

Я ищу что-то похожее на версию процессора, где вывод:

model.predict(data)

predict     False   True
False   0.316711    0.683289

Iхочу посмотреть доверительный интервал прогноза для GPU h2o. Что мне нужно сделать?

Answer 1

Я разобрался в проблеме. h2o4gpu использует ту же структуру, что и scikit-learn. я изменил прогноз на предикат_проба (данные) и вероятности были возвращены