gpt4 book ai didi

Python statsmodel.api 逻辑回归 (Logit)

转载 作者:太空宇宙 更新时间:2023-11-04 05:58:22 24 4
gpt4 key购买 nike

所以我正在尝试使用 python 的 statsmodels.api 对二元结果进行逻辑回归来进行预测。我按照教程使用 Logit。当我尝试对测试数据集进行预测时,每条记录的输出都是 0 到 1 之间的小数。它不应该给我零和一吗?还是我必须使用圆形函数或其他东西来转换它们。

请原谅这个问题的愚蠢。我正在凝视我的旅程。

最佳答案

预测值是给定解释变量的概率,更准确地说是观察到 1 的概率。

要获得 0、1 的预测,您需要选择一个阈值,例如 0.5 表示相等的阈值,并将 1 分配给高于阈值的概率。

对于 numpy 这将是例如

predicted = results.predict(x_for_prediction)
predicted_choice = (predicted > threshold).astype(int)

关于Python statsmodel.api 逻辑回归 (Logit),我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/26528019/

24 4 0
Copyright 2021 - 2024 cfsdn All Rights Reserved 蜀ICP备2022000587号
广告合作:1813099741@qq.com 6ren.com