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
• Guardrails - 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 AutoClassifier
from evalml.objectives import FraudCost


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

clf = AutoClassifier(objective=fraud_objective,
                     max_pipelines=3)

clf.fit(X_train, y_train)

clf.rankings

pipeline = clf.best_pipeline
pipeline.predict(X_test)

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

Read more about EvalML in our Documentation.