closek 1.1

Scikit-learn-style implementation of the close-k classifier.

Close-k Classifier

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.

Installation

Our package is available on PyPy, and can be installed using

pip install -i https://pypi.org/project/ closek

You can also install this package by cloning the Github repository, and running

pip install closek

If you want directly use the implementation in your package, you can also copy closek/closek.py into your code.

Usage

An example of how to use our package is shown in test.py.

Generating Results from Paper

The code used for the paper is in experiments. See the README there for more details.