Optimization-AI / LibAUC

LibAUC: A Deep Learning Library for X-Risk Optimization
https://libauc.org/
MIT License
273 stars 37 forks source link

关于09示例的技巧 #16

Closed yanruiD closed 2 years ago

yanruiD commented 2 years ago

我注意到在09的示例中,在论文的消融实验中,k>1时,对AUC的得分影响挺大。在代码里是如何展现调节k的值的呢,不太确定,望指教。谢谢大佬。

yzhuoning commented 2 years ago

目前的版本只提供了k=1的示例,k>1的话会在将来的更新中放出来

yanruiD commented 2 years ago

在论文里损失函数内部是w-α∇LAVG(w),这里的α在09示例里是如何调控的,代码里的alpha被绕晕了。谢谢,望指教。

yzhuoning commented 2 years ago

在09示例中,α的值和外部的lr一致,如果需要调节的话可以把lr0单独赋值并传到Optimizer里面即可。比如,可以让lr0比lr稍小一些。