From 895ff33965c444022ea6dc62140f5b4018a86dd6 Mon Sep 17 00:00:00 2001 From: Roque Lopez Date: Fri, 31 May 2024 17:17:33 -0400 Subject: [PATCH] Add list of possible primitive types --- alpha_automl/automl_api.py | 8 ++++++-- 1 file changed, 6 insertions(+), 2 deletions(-) diff --git a/alpha_automl/automl_api.py b/alpha_automl/automl_api.py index a261f89d..f9f7dc86 100644 --- a/alpha_automl/automl_api.py +++ b/alpha_automl/automl_api.py @@ -191,9 +191,13 @@ def get_pipeline(self, pipeline_id=None): def add_primitives(self, new_primitives): """ - Add new primitives. + Add new primitives to the search space. - :param new_primitives: Set of new primitives, tuples of name and object primitive + :param new_primitives: Set of new primitives, tuples of name and object primitive. Possible names are: + `IMPUTER`, `FEATURE_GENERATOR`, `FEATURE_SCALER`, `FEATURE_SELECTOR`, `TEXT_ENCODER`, `CATEGORICAL_ENCODER`, + `DATETIME_ENCODER`, `IMAGE_ENCODER`, `CLASSIFICATION_SINGLE_ENSEMBLER`, `CLASSIFICATION_MULTI_ENSEMBLER`, + `REGRESSION_SINGLE_ENSEMBLER`, `REGRESSION_MULTI_ENSEMBLER`, `CLASSIFIER`, `REGRESSOR`, `CLUSTERER`, + `TIME_SERIES_FORECASTER`, `SEMISUPERVISED_SELFTRAINER`, and `SEMISUPERVISED_LABELPROPAGATOR` """ for primitive_object, primitive_type in new_primitives: #check_input_for_multiprocessing(self._start_method, primitive_object, 'primitive')