src.gridmind.algorithms.evolutionary_rl.neuroevolution.neuroevolution_util

Classes

`NeuroEvolutionUtil`
`SimpleNN`

Module Contents

class src.gridmind.algorithms.evolutionary_rl.neuroevolution.neuroevolution_util.NeuroEvolutionUtil[source]

static get_parameters_vector(model)[source]

static set_parameters_vector(model, param_vector)[source]

static evaluate_fitness(env: gymnasium.Env, policy: gridmind.policies.parameterized.discrete_action_mlp_policy.DiscreteActionMLPPolicy, obs_preprocessor: Callable | None = None, average_over_episodes: int = 3)[source]

class src.gridmind.algorithms.evolutionary_rl.neuroevolution.neuroevolution_util.SimpleNN(input_size=4, hidden_size=10, output_size=2)[source]

Bases: torch.nn.Module

fc1[source]

fc2[source]

forward(x)[source]