Metadata-Version: 2.1
Name: evalml
Version: 0.43.0
Summary: EvalML is an AutoML library that builds, optimizes, and evaluates machine learning pipelines using domain-specific objective functions.
Home-page: https://github.com/alteryx/evalml/
Author: Alteryx, Inc.
Author-email: open_source_support@alteryx.com
License: UNKNOWN
Platform: UNKNOWN
Description-Content-Type: text/markdown
License-File: LICENSE
Requires-Dist: numpy (>=1.21.0)
Requires-Dist: pandas (<1.4.0,>=1.3.0)
Requires-Dist: scipy (>=1.5.0)
Requires-Dist: scikit-learn (>=0.24.0)
Requires-Dist: scikit-optimize (>=0.8.1)
Requires-Dist: pyzmq (>=20.0.0)
Requires-Dist: colorama (>=0.4.4)
Requires-Dist: cloudpickle (>=1.5.0)
Requires-Dist: click (>=7.1.2)
Requires-Dist: requirements-parser (>=0.2.0)
Requires-Dist: shap (>=0.36.0)
Requires-Dist: statsmodels (>=0.12.2)
Requires-Dist: texttable (>=1.6.2)
Requires-Dist: woodwork (>=0.11.0)
Requires-Dist: dask (>=2021.10.0)
Requires-Dist: nlp-primitives (>=2.1.0)
Requires-Dist: featuretools (>=1.2.0)
Requires-Dist: networkx (<2.6,>=2.5)
Requires-Dist: plotly (>=5.0.0)
Requires-Dist: kaleido (>=0.1.0)
Requires-Dist: ipywidgets (>=7.5)
Requires-Dist: xgboost (>=1.4.2)
Requires-Dist: catboost (>=0.20)
Requires-Dist: lightgbm (>=2.3.1)
Requires-Dist: matplotlib (>=3.3.3)
Requires-Dist: seaborn (>=0.11.1)
Requires-Dist: category-encoders (>=2.2.2)
Requires-Dist: imbalanced-learn (>=0.8.0)
Requires-Dist: pmdarima (>=1.8.1)
Requires-Dist: lime (>=0.2.0.1)
Requires-Dist: vowpalwabbit (>=8.11.0)
Requires-Dist: graphviz (>=0.13) ; platform_system != "Windows"
Requires-Dist: sktime (>=0.7.0) ; python_version < "3.9"
Provides-Extra: complete
Requires-Dist: alteryx-open-src-update-checker (>=2.0.0) ; extra == 'complete'
Requires-Dist: cmdstan-builder (==0.0.8) ; extra == 'complete'
Provides-Extra: prophet
Requires-Dist: cmdstan-builder (==0.0.8) ; extra == 'prophet'
Provides-Extra: update_checker
Requires-Dist: alteryx-open-src-update-checker (>=2.0.0) ; extra == 'update_checker'

<p align="center">
<img width=50% src="https://evalml-web-images.s3.amazonaws.com/evalml_horizontal.svg" alt="EvalML" />
</p>

<p align="center">
    <a href="https://github.com/alteryx/woodwork/actions?query=branch%3Amain+workflow%3ATests" target="_blank">
        <img src="https://github.com/alteryx/woodwork/workflows/Tests/badge.svg?branch=main" alt="Tests" />
    </a>
    <a href="https://codecov.io/gh/alteryx/evalml">
        <img src="https://codecov.io/gh/alteryx/evalml/branch/main/graph/badge.svg?token=JDc0Ib7kYL"/>
    </a>
    <a href="https://evalml.alteryx.com/en/latest/?badge=stable" target="_blank">
        <img src="https://readthedocs.com/projects/feature-labs-inc-evalml/badge/?version=stable" alt="Documentation Status" />
    </a>
    <a href="https://badge.fury.io/py/evalml" target="_blank">
        <img src="https://badge.fury.io/py/evalml.svg?maxAge=2592000" alt="PyPI Version" />
    </a>
    <a href="https://anaconda.org/conda-forge/evalml" target="_blank">
        <img src="https://anaconda.org/conda-forge/evalml/badges/version.svg" alt="Anaconda Version" />
    </a>
    <a href="https://pepy.tech/project/evalml" target="_blank">
        <img src="https://pepy.tech/badge/evalml/month" alt="PyPI Downloads" />
    </a>
</p>
<hr>

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.

## Installation 

Install from [PyPI](https://pypi.org/project/evalml/):

```bash
pip install evalml
```

or from the conda-forge channel on [conda](https://anaconda.org/conda-forge/evalml):

```bash
conda install -c conda-forge evalml
```

### Add-ons
**Update checker** - Receive automatic notifications of new Woodwork releases

PyPI:

```bash
pip install "evalml[update_checker]"
```
Conda:
```
conda install -c conda-forge alteryx-open-src-update-checker
```

## Start

#### Load and split example data 
```python
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
```python
from evalml.automl import AutoMLSearch
automl = AutoMLSearch(X_train=X_train, y_train=y_train, problem_type='binary')
automl.search()
```

#### View pipeline rankings
```python
automl.rankings
```

#### Get best pipeline and predict on new data
```python
pipeline = automl.best_pipeline
pipeline.predict(X_test)
```

## Next Steps

Read more about EvalML on our [documentation page](https://evalml.alteryx.com/):

* [Installation](https://evalml.alteryx.com/en/stable/install.html) and [getting started](https://evalml.alteryx.com/en/stable/start.html).
* [Tutorials](https://evalml.alteryx.com/en/stable/tutorials.html) on how to use EvalML.
* [User guide](https://evalml.alteryx.com/en/stable/user_guide.html) which describes EvalML's features.
* Full [API reference](https://evalml.alteryx.com/en/stable/api_reference.html)

## Built at Alteryx Innovation Labs
<a href="https://www.alteryx.com/innovation-labs">
    <img src="https://evalml-web-images.s3.amazonaws.com/alteryx_innovation_labs.png" alt="Alteryx Innovation Labs" />
</a>


