Skip to main content

Diego: Data IntElliGence Out.

Project description

Diego

Diego: Data in, IntElliGence Out.

简体中文

A fast framework that supports the rapid construction of automated learning tasks. Simply create an automated learning study (Study) and generate correlated trials (Trial). Then run the code and get a machine learning model. Implemented using Scikit-learn API glossary, using Bayesian optimization and genetic algorithms for automated machine learning.

Inspired by Fast.ai and MicroSoft nni.

Build Status PyPI GitHub GitHub code size in bytes

  • the classifier trained by a Study.
  • AutoML classifier with support for scikit-learn api. Support for exporting models and use them directly.
  • Hyperparametric optimization using Bayesian optimization and genetic algorithms
  • Supports bucketing/binning algorithm and LUS sampling method for preprocessing
  • Supports scikit-learn api classifier custom classifier for parameter search and super parameter optimization

Installation

You need to install swig first, and some rely on C/C++ interface compilation. Recommended to use conda installation

conda install --yes pip gcc swig libgcc=5.2.0
pip install diego

After installation, start with 6 lines of code to solve a machine learning classification problem.

Usage

Each task is considered to be a Study, and each Study consists of multiple Trial. It is recommended to create a Study first and then generate a Trial from the Study:

from diego.study import create_study
import sklearn.datasets
digits = sklearn.datasets.load_digits()
X_train, X_test, y_train, y_test = sklearn.model_selection.train_test_split(digits.data, digits.target,train_size=0.75, test_size=0.25)

s = create_study(X_train, y_train)
# can use default trials in Study

# or generate one
# s.generate_trials(mode='fast')
s.optimize(X_test, y_test)
# all_trials = s.get_all_trials()
# for t in all_trials:
#     print(t.__dict__)
#     print(t.clf.score(X_test, y_test))

RoadMap

ideas for releases in the future

  • 回归。
  • add documents.
  • 不同类型的Trial。TPE, BayesOpt, RandomSearch
  • 自定义的Trial。Trials by custom Classifier (like sklearn, xgboost)
  • 模型保存。model persistence
  • 模型输出。model output
  • basic Classifier
  • fix mac os hanged in optimize pipeline
  • add preprocessor
  • add FeatureTools for automated feature engineering

Project Structure

study, trials

Study:

Trial:

如果在OS X或者Linux多进程被 hang/crash/freeze

Since n_jobs>1 may get stuck during parallelization. Similar problems may occur in [scikit-learn] (https://scikit-learn.org/stable/faq.html#why-do-i-sometime-get-a-crash-freeze-with-n -jobs-1-under-osx-or-linux)

In Python 3.4+, one solution is to directly configure multiprocessing to use forkserver or spawn to start process pool management (instead of the default fork). For example, the forkserver mode is enabled globally directly in the code.

import multiprocessing
# other imports, custom code, load data, define model...
if __name__ == '__main__':
    multiprocessing.set_start_method('forkserver')

    # call scikit-learn utils with n_jobs > 1 here

more info :multiprocessing document

core

storage

For each study, the data storage and parameters, and the model is additionally stored in the Storage object, which ensures that Study only controls trials, and each Trial updates the results in the storage after updating, and updates the best results.

update result

When creating Study, you need to specify the direction of optimization maximize or minimize. Also specify the metrics for optimization when creating Trials. The default is maximize accuracy.

auto ml 补完计划

overview

bayes opt

  1. fmfn/bayes
  2. auto-sklearn

grid search

  1. H2O.ai

tree parzen

  1. hyperopt
  2. mlbox

metaheuristics grid search

  1. pybrain

generation

1.tpot

dl

  1. ms nni

issues

updates

TODO 文档更新。

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

diego-0.2.7.tar.gz (55.0 kB view details)

Uploaded Source

Built Distribution

If you're not sure about the file name format, learn more about wheel file names.

diego-0.2.7-py2.py3-none-any.whl (68.7 kB view details)

Uploaded Python 2Python 3

File details

Details for the file diego-0.2.7.tar.gz.

File metadata

  • Download URL: diego-0.2.7.tar.gz
  • Upload date:
  • Size: 55.0 kB
  • Tags: Source
  • Uploaded using Trusted Publishing? No
  • Uploaded via: twine/3.2.0 pkginfo/1.5.0.1 requests/2.23.0 setuptools/52.0.0 requests-toolbelt/0.9.1 tqdm/4.46.0 CPython/3.7.4

File hashes

Hashes for diego-0.2.7.tar.gz
Algorithm Hash digest
SHA256 a9338b4750a7b7e3d79f0bc9e6b403a84cccc455c002ad3ce0bdef6a118c301d
MD5 d9eb91d43ac274dd8ffc28419a05ec60
BLAKE2b-256 61b79f7bff11aeec39d4af912194c650107660e4a5ddbc3dee7d7fc65a799207

See more details on using hashes here.

File details

Details for the file diego-0.2.7-py2.py3-none-any.whl.

File metadata

  • Download URL: diego-0.2.7-py2.py3-none-any.whl
  • Upload date:
  • Size: 68.7 kB
  • Tags: Python 2, Python 3
  • Uploaded using Trusted Publishing? No
  • Uploaded via: twine/3.2.0 pkginfo/1.5.0.1 requests/2.23.0 setuptools/52.0.0 requests-toolbelt/0.9.1 tqdm/4.46.0 CPython/3.7.4

File hashes

Hashes for diego-0.2.7-py2.py3-none-any.whl
Algorithm Hash digest
SHA256 2ca0c5a168bfc98f91fce3fa5ef88afeabb599a70782441d6f9d17d2199a23ef
MD5 d9412131105ff86e67e191049d5be633
BLAKE2b-256 76bfd976ef2e4b017ff05d6a127c7f65fad0e50f5049c888ed6cec217d060826

See more details on using hashes here.

Supported by

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