From d811cfdbb7177c6d6ed8a147a6440dc94ec4e317 Mon Sep 17 00:00:00 2001 From: Hunter McGushion Date: Fri, 12 Jul 2019 22:50:37 -0700 Subject: [PATCH] Make `forge_experiment`'s `model_init_params` optional --- hyperparameter_hunter/optimization/protocol_core.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/hyperparameter_hunter/optimization/protocol_core.py b/hyperparameter_hunter/optimization/protocol_core.py index 61ad1935..ec5ffd11 100644 --- a/hyperparameter_hunter/optimization/protocol_core.py +++ b/hyperparameter_hunter/optimization/protocol_core.py @@ -216,7 +216,7 @@ def __init__( def forge_experiment( self, model_initializer, - model_init_params, + model_init_params=None, model_extra_params=None, feature_engineer=None, feature_selector=None, @@ -263,8 +263,8 @@ def forge_experiment( differences between calling :meth:`forge_experiment` and instantiating :class:`~hyperparameter_hunter.experiments.CVExperiment`""" self.model_initializer = model_initializer - self.model_init_params = identify_algorithm_hyperparameters(self.model_initializer) + model_init_params = model_init_params if model_init_params is not None else {} try: self.model_init_params.update(model_init_params) except TypeError: