PaddleFL
PaddleFL copied to clipboard
逻辑回归demo中的bug以及可能的解决方案
https://github.com/PaddlePaddle/PaddleFL/blob/08bfddb9160f9595b9ad9fb8563e16e39860e1ac/python/paddle_fl/mpc/examples/logistic_with_mnist/process_data.py#L142
逻辑回归模型的预测输出结果是1维的,在得到预测分类的时候却使用了argmax函数寻找多个维度中的最大值,这里输出的结果只能是0,这里存在bug。以sigmod二分类为例,我认为这里预测输出的结果应该是属于0-1之间的一个概率,大于0.5时结果分类为标签1,小于0.5时结果分类为标签0,应该改成如下这样:
if i > 0.5: f.write(str(1) + '\n') else : f.write(str(0) + '\n')