Update 03_Optimizing_AUPRC_with_ResNet18_on_Imbalanced_CIFAR10.py

yzhuoning · web-flow · commit 0d431510493f · 2021-12-10T00:32:55.000-06:00
diff --git a/examples/scripts/03_Optimizing_AUPRC_with_ResNet18_on_Imbalanced_CIFAR10.py b/examples/scripts/03_Optimizing_AUPRC_with_ResNet18_on_Imbalanced_CIFAR10.py
@@ -1,6 +1,7 @@
 """
 Author: Zhuoning Yuan, Qi Qi
 Contact: yzhuoning@gmail.com, qi-qi@uiowa.edu
+For more examples, please check Qi's repo: https://github.com/Optimization-AI/NeurIPS2021_SOAP
 """
 from libauc.losses import APLoss_SH
 from libauc.optimizers import SOAP_SGD
