`evorl.ec.optimizers.cem`¶

Module Contents¶

Classes¶

`SepCEM`	Sep Cross-Entropy Method.
`SepCEMState`	State of the SepCEM.

API¶

class evorl.ec.optimizers.cem.SepCEM[source]¶

Bases: evorl.ec.optimizers.ec_optimizer.EvoOptimizer

Sep Cross-Entropy Method.

ask(state: evorl.ec.optimizers.cem.SepCEMState) → tuple[chex.ArrayTree, evorl.ec.optimizers.ec_optimizer.ECState][source]¶

cov_eps_schedule: evorl.ec.optimizers.utils.ExponentialScheduleSpec¶: None

elite_weights: chex.Array¶: ‘pytree_field(…)’

init(mean: evorl.types.Params, key: chex.PRNGKey) → evorl.ec.optimizers.cem.SepCEMState[source]¶

mirror_sampling: bool¶: False

num_elites: int¶: None

pop_size: int¶: None

rank_weight_shift: float¶: 1.0

tell(state: evorl.ec.optimizers.cem.SepCEMState, fitnesses: chex.Array) → tuple[evorl.types.PyTreeDict, evorl.ec.optimizers.cem.SepCEMState][source]¶

weighted_update: bool¶: True

class evorl.ec.optimizers.cem.SepCEMState[source]¶

Bases: evorl.types.PyTreeData

State of the SepCEM.

cov_eps: chex.ArrayTree¶: None

key: chex.PRNGKey¶: None

mean: chex.ArrayTree¶: None

pop: None | chex.ArrayTree¶: None

variance: chex.ArrayTree¶: None