所以我尝试使用 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)
本文内容由网友自发贡献,版权归原作者所有,本站不承担相应法律责任。如您发现有涉嫌抄袭侵权的内容,请联系:hwhale#tublm.com(使用前将#替换为@)