Source code for

from typing import Optional

from autofit.database.sqlalchemy_ import sa
from autofit.non_linear.initializer import AbstractInitializer
from import AbstractPySwarms

[docs]class PySwarmsGlobal(AbstractPySwarms): __identifier_fields__ = ( "n_particles", "cognitive", "social", "inertia", ) def __init__( self, name: Optional[str] = None, path_prefix: Optional[str] = None, unique_tag: Optional[str] = None, initializer: Optional[AbstractInitializer] = 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, initializer=initializer, 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"] } filter_list = ["cognitive", "social", "inertia"] config_dict = {key: value for key, value in self.config_dict_search.items() if key not in filter_list} return pyswarms.global_best.GlobalBestPSO( dimensions=model.prior_count, bounds=bounds, init_pos=init_pos, options=options, **config_dict )