A classifier that endeavors to solve the saddle point problem for AUC maximization.
Project description
CalfMilp - A classifier with integer weights
A binomial classifier that fits using 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)
1.0
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 Distribution
calf_milp-0.0.3.tar.gz
(9.1 kB
view hashes)
Built Distribution
Close
Hashes for calf_milp-0.0.3-py3-none-any.whl
Algorithm | Hash digest | |
---|---|---|
SHA256 | 0dd415245d94850dad9baf5f7e30e55ca679068a3df43f93e52d3e28b72d1f7c |
|
MD5 | 28d1ecebf414e302276e3a9ff800ba45 |
|
BLAKE2b-256 | 9d9e91bb648f4db440c2152ad95d52d3553f007a2e7af5362c5503e96716ebf4 |