libauc 1.0.6

AUC Optimization

LibAUC

An end-to-end machine learning library for auc optimization.

Original Links

• Repository: https://github.com/yzhuoning/libauc
• Library website: https://libauc.org

How to install

$ pip install libauc

Example

Run the following commands

$ python

>>> from libauc import *
>>> ...
>>> Losss = AUCMLoss(imratio=0.1)
>>> optimizer = PESG(model, a=Loss.a, b=Loss.b, alpha=Loss.alpha, lr=0.1, gamma=500, weight_decay=1e-5)
>>> ...
>>> loss = Loss(y_pred, targets)
>>> optimizer.zero_grad()
>>> loss.backward(retain_graph=True)
>>> optimizer.step()

Maintainer

• Zhuoning Yuan [e-mail]