从 H2O 苏打水模型获取阶段结果

问题描述 投票:0回答:1

我希望为我的模型输出之一创建置信区间,我需要在应用链接函数之前获取模型输出。从我读到的内容来看,我似乎有兴趣获得模型的阶段结果。

到目前为止,我已经创建了一个具有适当参数的模型,对其进行了拟合,验证了参数值并获得了预测,但是无论如何我都没有在详细预测列中看到信息。

estimator = H2OGLMClassifier(family='binomial', featureCols=feature_columns, labelCol=response, withStageResults=True)
model = estimator.fit(training_data)

predictions = model.transform(training_data)

预测将有

detailed_predictions
列,但它只包含具有预测概率的 JSON,就像没有列出阶段结果一样,例如

{
  "label": "1",
  "probabilities": {"0": ".814", "1": ".176"}
}

是否还需要做些什么才能获得阶段性结果?阶段结果不是得到我想要的东西的正确方法吗?

谢谢

pyspark logistic-regression h2o sparkling-water h2o.ai
1个回答
0
投票

如果您希望获得结果的置信区间,那么您就是在寻找系数的置信区间。为此,您需要将 compute_p_values、remove_collinear_columns 设置为 true 来调用 GLM。模型构建过程完成后,您可以使用 model.coef_with_p_values() 返回模型系数、p 值和 std_error 以及其他字段。

© www.soinside.com 2019 - 2024. All rights reserved.