A lightweight automl framework
Project description
A lightweight automl framework for classification/regression tasks.
Installation
pip install cutoml
Usage Example
For classification,
from cutoml.cutoml import CutoClassifier
from sklearn.model_selection import train_test_split
from sklearn import datasets
dataset = datasets.load_digits()
X_train, X_test, y_train, y_test = train_test_split(dataset.data,
dataset.target,
test_size=0.2)
ctc = CutoClassifier(k_folds=3, n_jobs=-1, verbose=1)
ctc.fit(X=X_train, y=y_train)
For regression,
from cutoml.cutoml import CutoRegressor
from sklearn.model_selection import train_test_split
from sklearn import datasets
dataset = datasets.load_boston()
X_train, X_test, y_train, y_test = train_test_split(dataset.data,
dataset.target,
test_size=0.2)
ctr = CutoRegressor(k_folds=3, n_jobs=-1, verbose=1)
ctr.fit(X=X_train, y=y_train)
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
cutoml-0.0.6.tar.gz
(4.1 kB
view hashes)
Built Distribution
cutoml-0.0.6-py3-none-any.whl
(18.6 kB
view hashes)