o }oi@sddlmZddlmZddlmZddlmZddlmZddlm Z ddl m Z ddl m Z dd lmZdd lmZdd lmZdd lmZerPdd lmZGdddZdS)) annotations)Callable)Sequence)Any) TYPE_CHECKING)BaseDistribution)LazyRandomState)_constrained_dominates)perform_crossover) BaseCrossover) _dominates) FrozenTrial)Studyc@s(eZdZddddddZdddZdS)NSGAIIChildGenerationStrategyN) mutation_probconstraints_funcr float | None crossoverr crossover_probfloat swapping_probr/Callable[[FrozenTrial], Sequence[float]] | NonerngrreturnNonecCs|dusd|krdkstdtdd|kr#dks(tdtdd|kr5dks:tdtdt|tsGtd|d||_||_||_||_||_||_dS)Ng?zJ`mutation_prob` must be None or a float value within the range [0.0, 1.0].zC`crossover_prob` must be a float value within the range [0.0, 1.0].zB`swapping_prob` must be a float value within the range [0.0, 1.0].'zu' is not a valid crossover. For valid crossovers see https://optuna.readthedocs.io/en/stable/reference/samplers.html.) ValueError isinstancer _crossover_prob_mutation_prob_swapping_prob _crossover_constraints_func_rng)selfrrrrrrr&e/home/ubuntu/.local/lib/python3.10/site-packages/optuna/samplers/nsgaii/_child_generation_strategy.py__init__s4    z&NSGAIIChildGenerationStrategy.__init__studyr search_spacedict[str, BaseDistribution]parent_populationlist[FrozenTrial]dict[str, Any]c s|jdurtnt}|jj|jkr!t|j||||jj|j |}nt |}||jj |j fdd| D}t |}|jdurKdtd|}n|j}i} | D]} |jj|krd|| | | <qT| S)aGenerate a child parameter from the given parent population by NSGA-II algorithm. Args: study: Target study object. search_space: A dictionary containing the parameter names and parameter's distributions. parent_population: A list of trials that are selected as parent population. Returns: A dictionary containing the parameter names and parameter's values. Ncsi|]}||qSr&r&).0name parent_paramsr&r' Ysz:NSGAIIChildGenerationStrategy.__call__..r)r#r r r$rrandrr r"r!lenchoiceparamskeysr max) r%r)r*r, dominates child_paramsparent_population_sizen_paramsrr7 param_namer&r1r'__call__9s0    z&NSGAIIChildGenerationStrategy.__call__)rrrr rrrrrrrrrr)r)rr*r+r,r-rr.)__name__ __module__ __qualname__r(r?r&r&r&r'rs #rN) __future__rcollections.abcrrtypingrroptuna.distributionsr"optuna.samplers._lazy_random_stater.optuna.samplers.nsgaii._constraints_evaluationr !optuna.samplers.nsgaii._crossoverr (optuna.samplers.nsgaii._crossovers._baser optuna.study._multi_objectiver optuna.trialr optuna.studyrrr&r&r&r's