如何在H2O的自动ML函数中平衡类别?

3

请问在使用H2O的自动机器学习功能处理分类问题时,是否可以指定一些方法来平衡类别?

h2o.automl(x, y, training_frame, validation_frame = NULL,
leaderboard_frame = NULL, nfolds = 5, fold_column = NULL,
weights_column = NULL, max_runtime_secs = 3600, max_models = NULL,
stopping_metric = c("AUTO", "deviance", "logloss", "MSE", "RMSE", "MAE",
"RMSLE", "AUC", "lift_top_group", "misclassification",
"mean_per_class_error"), stopping_tolerance = NULL, stopping_rounds = 3,
seed = NULL, project_name = NULL) 

现有的文档似乎没有指定任何参数。谢谢。
1个回答

1

balance_classes选项在AutoML函数中未向用户公开(截至H2O v3.16.0.1),但是我们已经打开了一个工单,以在响应相对不平衡时(例如<10%的少数派类别)自动启用类平衡。 这应该很快就会添加。


您的需求是否仅需要在<10%时进行自动平衡,还是需要直接访问balance_classes(及相关)参数? - Erin LeDell
1
理想情况下,我希望采用后者,是否有时间表可以公布呢?感谢您的回复。 - elvikingo
1
我已经添加了一个工单,以完全暴露平衡类别参数,这应该很快完成:https://0xdata.atlassian.net/browse/PUBDEV-5377 - Erin LeDell
谢谢,期待这个版本的发布。 - elvikingo

网页内容由stack overflow 提供, 点击上面的
可以查看英文原文,
原文链接