EvalML is an AutoML library to build optimized machine learning pipelines for domain-specific objective functions.

Key Functionality

• Domain-specific - Includes repository of domain-specific objective functions and interface to define your own
• End-to-end - Constructs and optimizes pipelines that include imputation, feature selection, and a variety of modeling techniques
• Data Checks - Carefully cross-validates to prevent overfitting and warns you if training and testing results diverge

pip install evalml --index-url https://install.featurelabs.com/<KEY>

from evalml import AutoMLSearch
from evalml.objectives import FraudCost


fraud_objective = FraudCost(
    retry_percentage=.5,
    interchange_fee=.02,
    fraud_payout_percentage=.75,
    amount_col="amount"
)

automl = AutoMLSearch(problem_type='binary', objective=fraud_objective,
                     max_pipelines=3)

automl.search(X_train, y_train)

automl.rankings

pipeline = automl.best_pipeline
pipeline.predict(X_test)

• Configuring AutoClassifer
• Defining your own objective functions
• API Reference

Read more about EvalML in our Documentation.