Доступна ли средняя целевая кодировка H2O в Python?

Question

Я заметил, что H2O выпустила целевую среднюю кодировку

http://docs.h2o.ai/h2o/latest-stable/h2o-docs/data-munging/target-encoding.html

Он поставляется только с примером кода R. У кого-нибудь есть пример Python?

Answer 1

Как это:

from h2o.targetencoder import TargetEncoder

# Fit target encoding on training data
targetEncoder = TargetEncoder(x= ["addr_state", "purpose"], y = "bad_loan", fold_column = "cv_fold_te")
targetEncoder.fit(ext_train)

Но для этого требуется версия как минимум 3.22

Вот ссылка на пример: https://github.com/h2oai/h2o-tutorials/blob/78c3766741e8cbbbd8db04d54b1e34f678b85310/best-practices/feature-engineering/feature_engineering.ipynb

И ссылка на сам код: https://github.com/h2oai/h2o-3/blob/master/h2o-py/h2o/targetencoder.py