Skip to main content

A classifier that endeavors to solve the saddle point problem for AUC maximization.

Project description

CircleCI ReadTheDocs

CalfMilp

CalfMilp is a binomial classifier that implements a course approximation linear function by mixed integer linear programming.

Contact

Rolf Carlson hrolfrc@gmail.com

Install

Use pip to install calf-milp.

pip install calf-milp

Introduction

CalfMilp provides classification and prediction for two classes, the binomial case. Small problems are supported. This is research code and a work in progress.

CalfMilp is designed for use with scikit-learn pipelines and composite estimators.

Example

from calf_milp import CalfMilp
from sklearn.datasets import make_classification
from sklearn.model_selection import train_test_split
Make a classification problem
seed = 42
X, y = make_classification(
    n_samples=30,
    n_features=5,
    n_informative=2,
    n_redundant=2,
    n_classes=2,
    random_state=seed
)
X_train, X_test, y_train, y_test = train_test_split(X, y, random_state=seed)
Train the classifier
cls = CalfMilp().fit(X_train, y_train)
Get the score on unseen data
cls.score(X_test, y_test)
0.875

Project details


Download files

Download the file for your platform. If you're not sure which to choose, learn more about installing packages.

Source Distribution

calf_milp-0.1.8.tar.gz (9.2 kB view hashes)

Uploaded Source

Built Distribution

calf_milp-0.1.8-py3-none-any.whl (8.6 kB view hashes)

Uploaded Python 3

Supported by

AWS AWS Cloud computing and Security Sponsor Datadog Datadog Monitoring Fastly Fastly CDN Google Google Download Analytics Microsoft Microsoft PSF Sponsor Pingdom Pingdom Monitoring Sentry Sentry Error logging StatusPage StatusPage Status page