,YHhq|ddlmZddlZddlmZddlZddlmZddlm Z ddl m Z GddeejZ y) ) annotationsN)Any) BaseSampler) FrozenTrial) TrialStateceZdZdZdZdZdfd ZeddZeddZ ddZ e ddZ e jdd Z ejdd Zdd Zdd Zdd ZxZS) BaseGASamplerazBase class for Genetic Algorithm (GA) samplers. Genetic Algorithm samplers generate new trials by mimicking natural selection, using generations and populations to iteratively improve solutions. This base class defines the interface for GA samplers in Optuna and provides utility methods for managing generations and populations. The selection process is handled by :meth:`~BaseGASampler.select_parent`, which must be implemented by subclasses to define the parent selection strategy. Generation and population management is facilitated by methods like :meth:`~BaseGASampler.get_generation` and :meth:`~BaseGASampler.get_population`, ensuring consistent tracking and selection. Note: This class should be extended by subclasses that define specific GA sampling strategies, including parent selection and crossover operations. zBaseGASampler:generationzBaseGASampler:parent:c tt|di||jd|_|jd|_y)Nz :generationz:parent:)super__init_subclass____name___GENERATION_KEY_PARENT_CACHE_KEY_PREFIX)clskwargs __class__s R/var/www/html/planif/env/lib/python3.12/site-packages/optuna/samplers/_ga/_base.pyr zBaseGASampler.__init_subclass__$s9 !+F+!$k:*-,,x'@$c|jSN)rrs r_get_generation_keyz!BaseGASampler._get_generation_key)s"""rc|jSr)rrs r_get_parent_cache_key_prefixz*BaseGASampler._get_parent_cache_key_prefix-s+++rc||_yr_population_size)selfpopulation_sizes r__init__zBaseGASampler.__init__1s /rc|jSrr)rs rr zBaseGASampler.population_size4s$$$rc||_yrr)rvalues rr zBaseGASampler.population_size8s %rct)a9Select parent trials from the population for the given generation. This method is called once per generation to select parents from the population of the current generation. Output of this function is cached in the study system attributes. This method must be implemented in a subclass to define the specific selection strategy. Args: study: Target study object. generation: Target generation number. Returns: List of parent frozen trials. )NotImplementedError)rstudy generations r select_parentzBaseGASampler.select_parent<s ("!rc|jj|jd}||S|jdtj gd}d\}}t |D]A}|jj|jd}||kr3||kDr|}d}=|dz }C|jJd||jkr|}n|dz}|jj|j|j||S) abGet the generation number of the given trial. This method returns the generation number of the specified trial. If the generation number is not set in the trial's system attributes, it will calculate and set the generation number. The current generation number depends on the maximum generation number of all completed trials. Args: study: Study object which trial belongs to. trial: Trial object to get the generation number. Returns: Generation number of the given trial. NFTdeepcopystates use_cache)rrzPopulation size must be set.) system_attrsgetr _get_trialsrCOMPLETEreversedr_storageset_trial_system_attr _trial_id)rr'trialr(trialsmax_generationmax_generation_countts rget_trial_generationz"BaseGASampler.get_trial_generationRs&''++D,D,D,FM  ! ""E:;N;N:O[_"`/3,,&! *A++D,D,D,FKJN*n,!+'($$)$ *$$0P2PP0 $"7"7 7'J'!+J ,, OOT557 rc|jdtjgdDcgc]1}|jj |j d|k(r|3c}Scc}w)a Get the population of the given generation. Args: study: Target study object. generation: Target generation number. Returns: List of frozen trials in the given generation. FTr+N)r3rr4r1r2r)rr'r(r9s rget_populationzBaseGASampler.get_populationsj** (;(;'<+ !!%%d&>&>&@$G:U    s6Ac|dk(rgS|jj|j}|j|j t |zd}|%|j d}|Dcgc]}|| c}S|j||}|jj|j|j t |z|Dcgc]}|jc}|Scc}wcc}w)aGet the parent population of the given generation. This method caches the parent population in the study's system attributes. Args: study: Target study object. generation: Target generation number. Returns: List of parent frozen trials. If `generation == 0`, returns an empty list. rNF)r,) r6get_study_system_attrs _study_idr2rstrr3r)set_study_system_attrr8) rr'r(study_system_attrscached_parent_population_idsr:trial_idparent_populationr9s rget_parent_populationz#BaseGASampler.get_parent_populations ?I"^^BB5??S'9'='=  - - /#j/ A4( $ ( 3&&&6F5QRF8$R R $ 2 25* E  NN 0 0113c*oE.?@U@  % $S As 1 C-C2 )rr)returnrD)r int | None)rKrL)r$intrKNone)r' optuna.Studyr(rMrKzlist[FrozenTrial])r'rOr9rrKrM)r __module__ __qualname____doc__rrr classmethodrrr!propertyr setterabcabstractmethodr)r>r@rJ __classcell__)rs@rr r s&1O6A ##,,0%%&& ""*.` ( %rr ) __future__rrVtypingroptunaoptuna.samplers._baseroptuna.trial._frozenroptuna.trial._staterABCr r rrr`s."  -,*i%Ki%r