Source code for

from typing import Optional

from autofit.database.sqlalchemy_ import sa
from import AbstractPySwarms

[docs]class PySwarmsLocal(AbstractPySwarms): __identifier_fields__ = ( "n_particles", "cognitive", "social", "inertia", "number_of_k_neighbors", "minkowski_p_norm" ) def __init__( self, name: Optional[str] = None, path_prefix: Optional[str] = None, unique_tag: Optional[str] = None, iterations_per_update: int = None, number_of_cores: int = None, session: Optional[sa.orm.Session] = None, **kwargs ): """ A PySwarms Particle Swarm Optimizer global non-linear search. For a full description of PySwarms, checkout its Github and readthedocs webpages: Parameters ---------- name The name of the search, controlling the last folder results are output. path_prefix The path of folders prefixing the name folder where results are output. unique_tag The name of a unique tag for this model-fit, which will be given a unique entry in the sqlite database and also acts as the folder after the path prefix and before the search name. initializer Generates the initialize samples of non-linear parameter space (see autofit.non_linear.initializer). number_of_cores The number of cores sampling is performed using a Python multiprocessing Pool instance. """ super().__init__( name=name, path_prefix=path_prefix, unique_tag=unique_tag, iterations_per_update=iterations_per_update, number_of_cores=number_of_cores, session=session, **kwargs ) self.logger.debug("Creating PySwarms Search")
[docs] def search_internal_from(self, model, fitness, bounds, init_pos): """ Get the static Dynesty sampler which performs the non-linear search, passing it all associated input Dynesty variables. """ import pyswarms options = { "c1": self.config_dict_search["cognitive"], "c2": self.config_dict_search["social"], "w": self.config_dict_search["inertia"], "k": self.config_dict_search["number_of_k_neighbors"], "p": self.config_dict_search["minkowski_p_norm"], } filter_list = ["cognitive", "social", "inertia", "number_of_k_neighbors", "minkowski_p_norm"] config_dict = {key: value for key, value in self.config_dict_search.items() if key not in filter_list} return pyswarms.local_best.LocalBestPSO( dimensions=model.prior_count, bounds=bounds, init_pos=init_pos, options=options, **config_dict )