search

TsingZ0 / PFLlib

37 traditional FL (tFL) or personalized FL (pFL) algorithms, 3 scenarios, and 20 datasets.
GNU General Public License v2.0
1.36k stars 286 forks source link

AUC 的指标计算 #209

Open yu-4041 opened 2 days ago

yu-4041 commented 2 days ago

image 这里AUC指标的计算是不是存在问题？ 应该使用真值和概率值来计算AUC，但代码中使用的是模型输出的logits

TsingZ0 commented 21 hours ago

这里不是在计算AUC指标，这里只是统计真值和概率值。注意y_prob和y_true的命名

yu-4041 commented 19 hours ago

134行计算auc的输入y_prob和y_true。 对于y_prob应该使用softmax后的output，但代码没有进行softmax而是直接使用了output