Skip to main content

EvalML is an AutoML library that builds, optimizes, and evaluates machine learning pipelines using domain-specific objective functions.

Project description

Featuretools

GitHub Actions Codecov.io PyPI PyPI Stats

EvalML is an AutoML library which builds, optimizes, and evaluates machine learning pipelines using domain-specific objective functions.

Key Functionality

  • Automation - Makes machine learning easier. Avoid training and tuning models by hand. Includes data quality checks, cross-validation and more.
  • Data Checks - Catches and warns of problems with your data and problem setup before modeling.
  • End-to-end - Constructs and optimizes pipelines that include state-of-the-art preprocessing, feature engineering, feature selection, and a variety of modeling techniques.
  • Model Understanding - Provides tools to understand and introspect on models, to learn how they'll behave in your problem domain.
  • Domain-specific - Includes repository of domain-specific objective functions and an interface to define your own.

Install from PyPI

pip install evalml

Add-ons

Update checker
Receive automatic notifications of new EvalML releases

pip install evalml[update_checker]

Start

Load and split example data

import evalml
X, y = evalml.demos.load_breast_cancer()
X_train, X_test, y_train, y_test = evalml.preprocessing.split_data(X, y, problem_type='binary')

Run AutoML

from evalml.automl import AutoMLSearch
automl = AutoMLSearch(X_train=X_train, y_train=y_train, problem_type='binary')
automl.search()

View pipeline rankings

automl.rankings

Get best pipeline and predict on new data

pipeline = automl.best_pipeline
pipeline.predict(X_test)

Next Steps

Read more about EvalML on our documentation page:

Built at Alteryx Innovation Labs

Alteryx Innovation Labs

Project details


Download files

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

Files for evalml, version 0.29.0
Filename, size File type Python version Upload date Hashes
Filename, size evalml-0.29.0-py3-none-any.whl (6.2 MB) File type Wheel Python version py3 Upload date Hashes View
Filename, size evalml-0.29.0.tar.gz (3.2 MB) File type Source Python version None Upload date Hashes View

Supported by

AWS AWS Cloud computing Datadog Datadog Monitoring DigiCert DigiCert EV certificate Facebook / Instagram Facebook / Instagram PSF Sponsor Fastly Fastly CDN Google Google Object Storage and Download Analytics Microsoft Microsoft PSF Sponsor Pingdom Pingdom Monitoring Salesforce Salesforce PSF Sponsor Sentry Sentry Error logging StatusPage StatusPage Status page