This repository contains code accompanying
Minimizing Close-k Aggregate Loss Improves Classification
Bryan He, James Zou.
We provide a Python 3 implementation using the scikit-learn API, and provide code to reproduce the figures and tables from the paper.
The package can be installed using
pip install closek
If you want directly use the implementation in your package, you can also copy closek/closek.py into your code.
An example of how to use our package is shown in test.py.
The code used for the paper is in experiments. See the README there for more details.