AUC Optimization
Project description
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 yzhuoning@gmail.com
Project details
Release history Release notifications | RSS feed
Download files
Download the file for your platform. If you're not sure which to choose, learn more about installing packages.
Source Distributions
No source distribution files available for this release.See tutorial on generating distribution archives.
Built Distribution
libauc-1.0.5-py3-none-any.whl
(28.5 kB
view hashes)