replay_trajectory_classification.initial_conditions.UniformInitialConditions#

class UniformInitialConditions[source]#

Bases: object

Initial conditions where all discrete states and position bins are equally likely.

Methods

make_initial_conditions(environments, ...)

Creates initial conditions array

Methods

make_initial_conditions

Creates initial conditions array
make_initial_conditions(environments: tuple[Environment], environment_names_to_state: tuple[str]) → list[ndarray][source]#

Creates initial conditions array

Parameters:

  • environments (tuple[Environment]) – Spatial environments in the model

  • environment_names_to_state (tuple[str]) – Mapping of environment names to state

Returns:

initial_conditions

Return type:

list of arrays