Source code for evalml.pipelines.classification.catboost_binary

from evalml.pipelines import BinaryClassificationPipeline


[docs]class CatBoostBinaryClassificationPipeline(BinaryClassificationPipeline): """ CatBoost Pipeline for binary classification. CatBoost is an open-source library and natively supports categorical features. For more information, check out https://catboost.ai/ Note: impute_strategy must support both string and numeric data """ component_graph = ['Simple Imputer', 'CatBoost Classifier'] custom_hyperparameters = { "impute_strategy": ["most_frequent"], }