ForestOptimizer

class ForestOptimizer(search_space: dict[str, list], initialize: dict[Literal['grid', 'vertices', 'random', 'warm_start'], int | list[dict]] = None, constraints: list[callable] = None, random_state: int = None, rand_rest_p: float = 0, nth_process: int = None, boundary: str = 'clip', warm_start_smbo: pd.DataFrame | None = None, max_sample_size: int = 10000000, sampling: dict[Literal['random'], int] = None, replacement: bool = True, tree_regressor: Literal['random_forest', 'extra_tree', 'gradient_boost'] = 'extra_tree', tree_para: dict[str, int] = {'n_estimators': 100}, xi: float = 0.03)

Sequential model-based optimizer using tree ensemble surrogate models.

Forest Optimizer uses tree-based ensemble models (Random Forest or Extra Trees) as surrogate models instead of Gaussian Processes. This approach scales better to high-dimensional problems and large datasets while providing uncertainty estimates through the variance of tree predictions.

The algorithm follows the same sequential model-based optimization framework: (1) fit a tree ensemble to observed data, (2) use the ensemble to predict mean and variance at candidate points, (3) select the next point using an acquisition function, and (4) update the model with new observations.

The algorithm is well-suited for:

• High-dimensional optimization problems (>20 dimensions)

• Problems with many observations where GP fitting becomes slow

• Categorical or mixed parameter spaces

• Situations where tree-based models naturally fit the problem structure

Tree ensembles handle categorical variables naturally and can capture non-smooth objective functions better than GPs in some cases.

Parameters:

search_spacedict[str, list]

The search space to explore, defined as a dictionary mapping parameter names to arrays of possible values.

Each key is a parameter name (string), and each value is a numpy array or list of discrete values that the parameter can take. The optimizer will only evaluate positions that are on this discrete grid.

Example: A 2D search space with 100 points per dimension:

search_space = {
    "x": np.linspace(-10, 10, 100),
    "y": np.linspace(-10, 10, 100),
}

The resolution of each dimension (number of points in the array) directly affects optimization quality and speed. More points give finer resolution but increase the search space size exponentially.

initializedict[str, int], default={“vertices”: 4, “random”: 2}

Strategy for generating initial positions before the main optimization loop begins. Initialization samples are evaluated first, and the best one becomes the starting point for the optimizer.

Supported keys:

• "grid": int – Number of positions on a regular grid.

• "vertices": int – Number of corner/edge positions of the search space.

• "random": int – Number of uniformly random positions.

• "warm_start": list[dict] – Specific positions to evaluate, each as a dict mapping parameter names to values.

Multiple strategies can be combined:

initialize = {"vertices": 4, "random": 10}
initialize = {"warm_start": [{"x": 0.5, "y": 1.0}], "random": 5}

More initialization samples improve the starting point but consume iterations from n_iter. For expensive objectives, a few targeted warm-start points are often more efficient than many random samples.

constraintslist[callable], default=[]

A list of constraint functions that restrict the search space. Each constraint is a callable that receives a parameter dictionary and returns True if the position is valid, False if it should be rejected.

Rejected positions are discarded and regenerated: the optimizer resamples a new candidate position (up to 100 retries per step). During initialization, positions that violate constraints are filtered out entirely.

Example: Constrain the search to a circular region:

def circular_constraint(para):
    return para["x"]**2 + para["y"]**2 <= 25

constraints = [circular_constraint]

Multiple constraints are combined with AND logic (all must return True).

random_stateint or None, default=None

Seed for the random number generator to ensure reproducible results.

• None: Use a new random state each run (non-deterministic).

• int: Seed the random number generator for reproducibility.

Setting a fixed seed is recommended for debugging and benchmarking. Different seeds may lead to different optimization trajectories, especially for stochastic optimizers.

rand_rest_pfloat, default=0

Probability of performing a random restart instead of the normal algorithm step. At each iteration, a uniform random number is drawn; if it falls below rand_rest_p, the optimizer jumps to a random position instead of following its strategy.

• 0.0: No random restarts (pure algorithm behavior).

• 0.01-0.05: Light diversification, helps escape shallow local optima.

• 0.1-0.3: Aggressive restarts, useful for highly multi-modal landscapes.

• 1.0: Equivalent to random search.

This is especially useful for local search optimizers (Hill Climbing, Simulated Annealing) that can get trapped. For population-based optimizers, the effect is less pronounced since they already maintain diversity through multiple agents.

boundary{“clip”, “reflect”, “periodic”, “random”, “intermediate”}, default=”clip”

Strategy for handling positions that exceed the search space bounds. When the optimizer proposes a candidate outside the valid range, this parameter controls how that candidate is mapped back.

• "clip": Clamp each coordinate to the nearest bound.

• "reflect": Mirror the position back into the search space at the boundary it crossed.

• "periodic": Wrap around to the opposite end of the range, treating the search space as periodic.

• "random": Replace the out-of-bounds position with a uniformly random position within the valid range.

• "intermediate": Move to the midpoint between the current position and the violated bound.

Applies to continuous and discrete numerical dimensions. Categorical dimensions always snap to the nearest valid category index.

warm_start_smboobject or None, default=None

Previous SMBO state for warm-starting the surrogate model. Allows continuing optimization from a previous run by reusing the fitted model state, avoiding the cost of refitting from scratch.

Pass None to start fresh without warm-starting.

max_sample_sizeint, default=10000000

Maximum number of candidate points to consider when optimizing the acquisition function. The surrogate model predicts scores for these candidates, and the best one according to the acquisition function is selected for evaluation.

Larger values improve acquisition optimization quality but increase memory usage and computation time. For most problems, the default is more than sufficient. Reduce this if memory is a concern.

samplingdict, default={“random”: 1000000}

Configuration for how candidate points are generated for acquisition function optimization. The key specifies the sampling strategy and the value specifies the number of samples.

Currently supported: {"random": N} for uniform random sampling of N candidate points from the search space.

Example:

sampling = {"random": 500000}  # Fewer candidates, faster

replacementbool, default=True

Whether to sample candidate points with replacement when generating candidates for acquisition function optimization. When True, the same point can appear multiple times. When False, each candidate is unique, ensuring diversity but potentially slower for very large sample sizes.

tree_regressor{“extra_tree”, “random_forest”}, default=”extra_tree”

The type of tree ensemble used as the surrogate model.

• "extra_tree": Extra-Trees (Extremely Randomized Trees). Faster to train because it uses random split thresholds instead of searching for optimal splits. Provides smoother uncertainty estimates. Recommended for most cases.

• "random_forest": Standard Random Forest. Uses optimal split searching, which can provide more accurate predictions but is slower to train.

tree_paradict, default={“n_estimators”: 100}

Parameters passed directly to the underlying scikit-learn tree regressor. Common options include:

• n_estimators: Number of trees in the ensemble. More trees provide better uncertainty estimates but increase training time. Default is 100.

• max_depth: Maximum tree depth. Shallower trees provide smoother predictions.

• min_samples_split: Minimum samples to split a node.

Example:

tree_para = {"n_estimators": 200, "max_depth": 10}

xifloat, default=0.03

Exploration-exploitation trade-off parameter for the acquisition function. Controls how much the optimizer values uncertain regions (high variance across trees) over predicted-good regions.

• 0.0: Pure exploitation, samples where the ensemble predicts the best score.

• 0.01-0.05: Mild exploration (default region).

• 0.1-0.3: Moderate exploration, favors uncertain regions.

Same role as xi in BayesianOptimizer, but uncertainty is estimated from the variance across tree predictions rather than from a Gaussian Process.

Attributes:

best_para

Return the best parameters found as a dictionary.

best_score

Best score found during the search.

best_value

Return the best values found (raw parameter values).

diagnostics

Diagnostics accessor for the last search.

search_data

Lazily construct and return the search results DataFrame.

Methods

search(objective_function, n_iter[, ...])

Run the optimization loop.

eval_time
init_stats
iter_time

See also

BayesianOptimizer

SMBO using Gaussian Processes.

TreeStructuredParzenEstimators

SMBO using density ratio estimation.

EnsembleOptimizer

Combines multiple surrogate model types.

Notes

The algorithm follows the same SMBO framework as Bayesian Optimization but replaces the Gaussian Process with a tree ensemble:

1. Fit a tree ensemble (Extra-Trees or Random Forest) to all observed (position, score) pairs.

2. For each candidate point, predict mean (average across trees) and uncertainty (variance across trees).

3. Select the next point using the acquisition function weighted by xi.

Tree ensembles offer several advantages over GPs:

• Scalability: Training is \(O(n \\log n)\) vs. \(O(n^3)\) for GPs.

• Categorical support: Trees handle categorical features natively.

• Non-stationarity: Trees can model functions with varying smoothness across the space.

The trade-off is that tree-based uncertainty estimates are less principled than GP posterior variance.

For visual explanations and tuning guides, see the Forest Optimizer user guide.

Examples

>>> import numpy as np
>>> from gradient_free_optimizers import ForestOptimizer

>>> def high_dim_function(para):
...     return -sum(para[f"x{i}"] ** 2 for i in range(5))

>>> search_space = {f"x{i}": np.linspace(-5, 5, 50) for i in range(5)}

>>> opt = ForestOptimizer(
...     search_space,
...     tree_regressor="extra_tree",
...     tree_para={"n_estimators": 50},
... )
>>> opt.search(high_dim_function, n_iter=200)

search(objective_function: Callable[[dict[str, Any]], float], n_iter: int, max_time: float | None = None, max_score: float | None = None, early_stopping: dict[str, Any] | None = None, memory: bool | BaseStorage = True, memory_warm_start: pd.DataFrame | None = None, verbosity: list[str] | Literal[False] = ['progress_bar', 'print_results', 'print_times'], optimum: Literal['maximum', 'minimum'] = 'maximum', callbacks: list[Callable[[CallbackInfo], bool | None]] | None = None, catch: dict[type[Exception], int | float] | None = None) → None

Run the optimization loop.

Evaluates objective_function up to n_iter times, searching for the parameters that maximize (or minimize) the returned score. The search proceeds in two phases: an initialization phase that evaluates starting positions (controlled by the initialize constructor parameter), followed by an iteration phase where the optimizer’s strategy generates new candidate positions.

After the search finishes, results are available via best_para, best_score, and search_data.

Parameters:

objective_functioncallable

The function to optimize. Must accept a single dictionary mapping parameter names to values and return either:

• A float score, or

• A tuple (float, dict) where the second element contains custom metrics (accessible via callbacks and search_data).

Example:

def objective(params):
    return -(params["x"] ** 2 + params["y"] ** 2)

def objective_with_metrics(params):
    loss = params["x"] ** 2
    return -loss, {"loss": loss}

n_iterint

Total number of iterations (including initialization). Each iteration evaluates the objective function once (unless a cached result is found when memory=True).

max_timefloat or None, default=None

Maximum wall-clock time in seconds. The search stops after the current iteration if the elapsed time exceeds this limit. None means no time limit.

max_scorefloat or None, default=None

Target score threshold. The search stops when the best score found so far reaches or exceeds this value. When optimum="minimum", this refers to the original (non-negated) score. None means no score limit.

early_stoppingdict or None, default=None

Configuration for stopping the search when progress stalls. None disables early stopping. When provided, the dictionary supports the following keys:

• "n_iter_no_change" (int, required): Stop if no improvement is observed for this many consecutive iterations.

• "tol_abs" (float, optional): Minimum absolute improvement required over the window to count as progress.

• "tol_rel" (float, optional): Minimum relative improvement (in percent) required over the window to count as progress.

Example:

early_stopping = {"n_iter_no_change": 50}
early_stopping = {"n_iter_no_change": 30, "tol_abs": 0.001}

memorybool or BaseStorage, default=True

Controls evaluation caching. When True, uses an in-memory dictionary (equivalent to MemoryStorage()). When False, disables caching entirely. A BaseStorage instance enables custom storage backends:

from gradient_free_optimizers._storage import SQLiteStorage
opt.search(objective, memory=SQLiteStorage("results.db"))

SQLiteStorage persists results to disk, enabling crash recovery and cache reuse across runs. Works with distributed evaluation (positions are checked against the cache before being dispatched to workers).

In-memory caching is especially useful for discrete search spaces where revisits are common.

memory_warm_startpd.DataFrame or None, default=None

A DataFrame from a previous search (typically obtained via search_data) to pre-populate the evaluation cache. The DataFrame must contain columns matching the search space parameter names plus a "score" column. Requires memory=True.

Example:

opt1 = HillClimbingOptimizer(search_space)
opt1.search(objective, n_iter=50)

opt2 = HillClimbingOptimizer(search_space)
opt2.search(objective, n_iter=50,
            memory_warm_start=opt1.search_data)

verbositylist[str] or False, default=[“progress_bar”, “print_results”, “print_times”]

Controls console output during and after the search. Pass False or an empty list for silent operation.

Supported values:

• "progress_bar": Show a live tqdm progress bar during the search.

• "print_results": Print best score and best parameters after the search completes.

• "print_times": Print timing breakdown (evaluation time, optimization overhead, throughput) after the search completes.

• "print_search_stats": Print search statistics including iteration counts, acceptance rate, number of improvements, and longest plateau.

• "print_statistics": Print score statistics (min, max, mean, std) after the search completes.

• "debug_stop": Print detailed stopping condition debug info when the search terminates early.

optimum{“maximum”, “minimum”}, default=”maximum”

Whether to maximize or minimize the objective function. When set to "minimum", the objective function’s return value is negated internally so that the optimizer always maximizes. The reported best_score is in original (non-negated) units.

callbackslist[callable] or None, default=None

A list of callback functions invoked after each iteration. Each callback receives a single argument info with the following attributes:

• info.iteration (int): Current iteration index (0-based).

• info.score (float): Score from the current evaluation.

• info.params (dict): Parameters evaluated in this iteration.

• info.best_score (float): Best score found so far.

• info.best_para (dict): Parameters of the best score.

• info.n_iter (int): Total iterations planned.

• info.phase (str): "init" or "iter".

• info.elapsed_time (float): Seconds since search started.

• info.metrics (dict): Custom metrics from the objective function (empty if the objective returns only a score).

• info.convergence (list[float]): Best score at each iteration so far.

If any callback returns False, the search stops immediately. Any other return value (including None) is ignored and the search continues.

Example:

def log_progress(info):
    if info.iteration % 10 == 0:
        print(f"Iter {info.iteration}: best={info.best_score:.4f}")

def stop_early(info):
    if info.best_score > 0.99:
        return False  # stops the search

opt.search(objective, n_iter=100,
           callbacks=[log_progress, stop_early])

catchdict[type, float] or None, default=None

Error handling for the objective function. Maps exception types to fallback scores. When the objective function raises a caught exception, the optimizer records the fallback score instead of crashing. Exception subclasses are matched via isinstance, so {Exception: ...} catches all.

The fallback score is in the user’s original units (before any negation from optimum="minimum"). Use float('nan') or float('inf') to mark positions as invalid without inventing an artificial score.

Example:

catch = {ValueError: -1000, RuntimeError: float('nan')}

opt.search(objective, n_iter=100, catch=catch)

Examples

Basic usage with default settings:

>>> import numpy as np
>>> from gradient_free_optimizers import HillClimbingOptimizer
>>> def objective(para):
...     return -(para["x"] ** 2)
>>> search_space = {"x": np.linspace(-10, 10, 100)}
>>> opt = HillClimbingOptimizer(search_space)
>>> opt.search(objective, n_iter=30)

Using multiple stopping conditions:

>>> opt.search(
...     objective,
...     n_iter=1000,
...     max_time=60,
...     max_score=-0.01,
...     early_stopping={"n_iter_no_change": 50},
... )

property best_para

Return the best parameters found as a dictionary.

Resolution order: 1. Explicitly set _best_para (used by the ask/tell mixin). 2. SearchTracker-derived value when search() has populated it. 3. Fallback computed from _pos_best.

The tracker path is preferred because _pos_best can lag behind the true best for some optimizers that only update it on accepted moves. In v2 the fallback path is slated for removal; the tracker becomes the single source of truth.

property best_score: float

Best score found during the search.

Reads from the internal SearchTracker (the single source of truth) via the private self._data accessor.

property best_value

Return the best values found (raw parameter values).

Returns:

list or None

List of best values in parameter order, or None if no evaluation has been performed yet.

property diagnostics

Diagnostics accessor for the last search.

Returns an accessor whose methods run the diagnostics in gradient_free_optimizers.diagnostics on this optimizer’s search_data.

For saved runs or cross-run analysis, prefer the free functions in gradient_free_optimizers.diagnostics which accept any list-of-dicts, pandas/polars DataFrame, or dict-of-sequences.

Raises:

AttributeError

If search() has not been called yet.

property search_data: pd.DataFrame

Lazily construct and return the search results DataFrame.

The DataFrame is only built when this property is accessed, avoiding a large memory spike at the end of high-dimensional optimizations. The result is cached so subsequent accesses don’t rebuild it.