【问题标题】:Why does H2O Stacked ensemble return only a single class probability?为什么 H2O Stacked ensemble 只返回一个类概率?
【发布时间】:2020-11-08 09:33:54
【问题描述】:

我注意到,与返回两个类别概率(p0 和 p1)的 H2O GBM 不同,H2O Stacked ensemble 仅返回单个类别概率。谁能解释一下为什么会这样?这个概率是否默认对应于 p1?

H2O 堆叠系综预测:

H2O GBM 模型预测:

【问题讨论】:

    标签: machine-learning h2o


    【解决方案1】:

    您一定是不小心使用 Stacked Ensemble 训练了一个回归模型(回归预测只是一列)。对于 H2O 进行分类,您需要确保您的响应列是“因子”(又名“枚举”)类型,否则 H2O 将进行回归。

    【讨论】:

    • 我已经使用 asfactor() 将目标二进制变量转换为因子。由于这是一个分类问题,我使用 H2ORandomForestEstimator 和 H2OXGBoostEstimator 作为我的基本估计器,并使用 H2OGradientBoostingEstimator 作为 H2OStackedEnsembleEstimator 对象中的元估计器。
    • 我发现了这个问题。我错误地将不同的目标列转换为因子类型。感谢您的澄清。
    猜你喜欢
    • 1970-01-01
    • 1970-01-01
    • 1970-01-01
    • 1970-01-01
    • 1970-01-01
    • 2017-11-29
    • 1970-01-01
    • 2017-11-27
    • 2016-08-18
    相关资源
    最近更新 更多