API Reference

Demo Datasets

load_fraud	Load credit card fraud dataset.
load_wine	Load wine dataset.
load_breast_cancer	Load breast cancer dataset.
load_diabetes	Load diabetes dataset.

Preprocessing

load_data	Load features and labels from file(s).
split_data	Splits data into train and test sets.

AutoML

AutoClassificationSearch	Automatic pipeline search class for classification problems
AutoRegressionSearch	Automatic pipeline search for regression problems

Plotting

AutoClassificationSearch.plot.get_roc_data	Gets data that can be used to create a ROC plot.
AutoClassificationSearch.plot.generate_roc_plot	Generate Receiver Operating Characteristic (ROC) plot for a given pipeline using cross-validation using the data returned from get_roc_data().
AutoRegressionSearch.plot.get_roc_data	Gets data that can be used to create a ROC plot.
AutoRegressionSearch.plot.generate_roc_plot	Generate Receiver Operating Characteristic (ROC) plot for a given pipeline using cross-validation using the data returned from get_roc_data().
AutoClassificationSearch.plot.get_confusion_matrix_data	Gets data that can be used to create a confusion matrix plot.
AutoClassificationSearch.plot.generate_confusion_matrix	Generate confusion matrix plot for a given pipeline using the data returned from get_confusion_matrix_data().
AutoRegressionSearch.plot.get_confusion_matrix_data	Gets data that can be used to create a confusion matrix plot.
AutoRegressionSearch.plot.generate_confusion_matrix	Generate confusion matrix plot for a given pipeline using the data returned from get_confusion_matrix_data().

Model Types

list_model_types

List model type for a particular problem type

Components

Transformers

OneHotEncoder	Creates one-hot encoding for non-numeric data
RFRegressorSelectFromModel	Selects top features based on importance weights using a Random Forest regressor
RFClassifierSelectFromModel	Selects top features based on importance weights using a Random Forest classifier
SimpleImputer	Imputes missing data with either mean, median and most_frequent for numerical data or most_frequent for categorical data
StandardScaler	Standardize features: removes mean and scales to unit variance

Estimators

LogisticRegressionClassifier	Logistic Regression Classifier
RandomForestClassifier	Random Forest Classifier
XGBoostClassifier	XGBoost Classifier
LinearRegressor	Linear Regressor
RandomForestRegressor	Random Forest Regressor

Pipelines

get_pipelines	Returns potential pipelines by model type
save_pipeline	Saves pipeline at file path
load_pipeline	Loads pipeline at file path

PipelineBase
RFClassificationPipeline	Random Forest Pipeline for both binary and multiclass classification
XGBoostPipeline	XGBoost Pipeline for both binary and multiclass classification
LogisticRegressionPipeline	Logistic Regression Pipeline for both binary and multiclass classification
RFRegressionPipeline	Random Forest Pipeline for regression problems
LinearRegressionPipeline	Linear Regression Pipeline for regression problems

Plotting

PipelineBase.plot

Objective Functions

Domain Specific

FraudCost	Score the percentage of money lost of the total transaction amount process due to fraud
LeadScoring	Lead scoring

Classification

F1	F1 score for binary classification
F1Micro	F1 score for multiclass classification using micro averaging
F1Macro	F1 score for multiclass classification using macro averaging
F1Weighted	F1 score for multiclass classification using weighted averaging
Precision	Precision score for binary classification
PrecisionMicro	Precision score for multiclass classification using micro averaging
PrecisionMacro	Precision score for multiclass classification using macro averaging
PrecisionWeighted	Precision score for multiclass classification using weighted averaging
Recall	Recall score for binary classification
RecallMicro	Recall score for multiclass classification using micro averaging
RecallMacro	Recall score for multiclass classification using macro averaging
RecallWeighted	Recall score for multiclass classification using weighted averaging
AUC	AUC score for binary classification
AUCMicro	AUC score for multiclass classification using micro averaging
AUCMacro	AUC score for multiclass classification using macro averaging
AUCWeighted	AUC Score for multiclass classification using weighted averaging
LogLoss	Log Loss for both binary and multiclass classification
MCC	Matthews correlation coefficient for both binary and multiclass classification
ROC	Receiver Operating Characteristic score for binary classification.
ConfusionMatrix	Confusion matrix for classification problems

Regression

R2	Coefficient of determination for regression
MAE	Mean absolute error for regression
MSE	Mean squared error for regression
MSLE	Mean squared log error for regression
MedianAE	Median absolute error for regression
MaxError	Maximum residual error for regression
ExpVariance	Explained variance score for regression

Problem Types

ProblemTypes

Enum for type of machine learning problem: BINARY, MULTICLASS, or REGRESSION

handle_problem_types

Handles problem_type by either returning the ProblemTypes or converting from a str

Tuners

Tuner	Defines API for Tuners
SKOptTuner	Bayesian Optimizer

Guardrails

detect_highly_null	Checks if there are any highly-null columns in a dataframe.
detect_label_leakage	Check if any of the features are highly correlated with the target.
detect_outliers	Checks if there are any outliers in a dataframe by using first Isolation Forest to obtain the anomaly score of each index and then using IQR to determine score anomalies.
detect_id_columns	Check if any of the features are ID columns.