【发布时间】:2021-05-05 16:01:11
【问题描述】:
我在运行 h2o AutoML 时收到此警告。我安装了 3.32.1.2 版本,并在 python 3.8 上运行它。
AutoML progress: |
11:30:52.773: AutoML: XGBoost is not available; skipping it.
代码:
import h2o
h2o.init()
h2o_df = h2o.H2OFrame(df)
train, test = h2o_df.split_frame(ratios=[.75])
# Identify predictors and response
x = train.columns
y = "TERM_DEPOSIT"
x.remove(y)
from h2o.automl import H2OAutoML
aml = H2OAutoML(max_runtime_secs=600,
#exclude_algos=['DeepLearning'],
seed=1,
#stopping_metric='logloss',
#sort_metric='logloss',
balance_classes=False,
project_name='Completed'
)
%time aml.train(x=x, y=y, training_frame=train)
【问题讨论】:
-
嗨 Leo,你也可以试试 MLJAR AutoML,它是一个开源的 python 包,它有 xgboost、lightgbm 和 catboost 算法(来自 GBM 家族)github.com/mljar/mljar-supervised
标签: python-3.x xgboost h2o automl