examples.systems.parameters
HYPERPARAMETERS: Final[neural_dynamics.core.hyperparameters.Hyperparameters] =
Hyperparameters(drift_network=NetworkArchitecture(input_size=1, hidden_sizes=[100, 100], output_size=1), diffusion_network=NetworkArchitecture(input_size=1, hidden_sizes=[100, 100], output_size=1), critic_network=NetworkArchitecture(input_size=1, hidden_sizes=[100, 100], output_size=1), state_dimension=1, input_dimension=1, hidden_variables=0, external_hidden_variables=0, timestep=0.01, learning_rates=LearningRates(drift=0.001, diffusion=0.001, critic=0.001, generator=0.001), number_of_epochs=100, number_of_gan_epochs=100, critic_updates=5, gradient_penalty_weight=10.0, batch_size=64, sde_l1_weight=0.1, moment_matching_weight=0.0, moment_matching_enabled=False, train_ode_with_sde=False, drift_l1_weight=1.0)
DUFFING_OSCILLATOR_PARAMETERS: Final[examples.systems.parameters.duffing_oscillator.DuffingOscillatorParameters] =
DuffingOscillatorParameters(alpha=-1.0, beta=1.0, mu=0.15, gamma=0.4, Omega=0.9, Theta=4.141947e-21, D_tilde=0.04, t_correl_tilde=0.5, timestep=0.005, total_time=25.0, initial_position=0.1, initial_velocity=0.0, initial_coloured_noise=0.0)
def
get_parameters( system: str) -> Union[neural_dynamics.core.hyperparameters.Hyperparameters, examples.systems.parameters.duffing_oscillator.DuffingOscillatorParameters]: