diff --git a/bamt/builders/composite_builder.py b/bamt/builders/composite_builder.py index bde33b2..eb67562 100644 --- a/bamt/builders/composite_builder.py +++ b/bamt/builders/composite_builder.py @@ -187,21 +187,24 @@ def search(self, data: DataFrame, **kwargs) -> [List[Tuple[str, str]], Dict]: preprocessed_data, _ = p.apply(data) # Create the initial population - initial = [ - CompositeModel( - nodes=[ - CompositeNode( - nodes_from=None, - content={ - "name": vertex, - "type": p.nodes_types[vertex], - "parent_model": None, - }, - ) - for vertex in vertices - ] - ) - ] + initial = kwargs.get( + "custom_initial_structure", + [ + CompositeModel( + nodes=[ + CompositeNode( + nodes_from=None, + content={ + "name": vertex, + "type": p.nodes_types[vertex], + "parent_model": None, + }, + ) + for vertex in vertices + ] + ) + ], + ) # Define the requirements for the evolutionary algorithm requirements = GraphRequirements( diff --git a/bamt/builders/evo_builder.py b/bamt/builders/evo_builder.py index 0b7c39d..d8bc92e 100644 --- a/bamt/builders/evo_builder.py +++ b/bamt/builders/evo_builder.py @@ -79,7 +79,6 @@ def __init__( self.default_timeout = 180 self.default_num_of_generations = 50 self.default_early_stopping_iterations = 50 - self.verbose = (True,) self.logging_level = 50 self.objective_metric = evo.K2_metric self.default_crossovers = [ @@ -98,6 +97,7 @@ def __init__( has_no_cycle, evo.has_no_duplicates, ] + self.verbose = True def build( self,