SimFish

• simfish_rl package
  • Submodules
  • simfish_rl.eval_simfish_agent module
    • StateEnvLoop
      • StateEnvLoop.run_episode()
    • eval_agent()
  • simfish_rl.hdf5_logger module
    • EnvInfoKeep
      • EnvInfoKeep.get_metrics()
      • EnvInfoKeep.observe()
      • EnvInfoKeep.observe_first()
    • HDF5Logger
      • HDF5Logger.close()
      • HDF5Logger.write()
    • SimpleEnvInfoKeep
      • SimpleEnvInfoKeep.get_metrics()
      • SimpleEnvInfoKeep.observe()
      • SimpleEnvInfoKeep.observe_first()
  • simfish_rl.simfish_r2d2_learner module
    • SimfishR2D2Learner
      • SimfishR2D2Learner.step()
    • reflect_actions()
    • reflect_observations()
    • reflect_samples()
  • simfish_rl.simfish_r2d2_network module
    • DeepSimfishTorso
    • Flatten
    • OAREmbedding
      • OAREmbedding.num_actions
      • OAREmbedding.torso
    • R2D2SimfishNetwork
      • R2D2SimfishNetwork._embed
      • R2D2SimfishNetwork._core
      • R2D2SimfishNetwork._duelling_head
      • R2D2SimfishNetwork._num_actions
      • R2D2SimfishNetwork.__call__()
      • R2D2SimfishNetwork.initial_state()
      • R2D2SimfishNetwork.unroll()
      • R2D2SimfishNetwork.initial_state()
      • R2D2SimfishNetwork.unroll()
    • make_r2d2_networks()
    • retina
  • Module contents
• simulation package
  • Submodules
  • simulation.arena module
    • Arena
      • Arena.test_mode
      • Arena.arena_width
      • Arena.arena_height
      • Arena.bottom_intensity
      • Arena.dark_gain
      • Arena.light_gradient
      • Arena.dark_light_ratio
      • Arena.sediment_sigma
      • Arena.max_uv_range
      • Arena.max_red_range
      • Arena.global_sediment_grating
      • Arena.global_luminance_mask
      • Arena.illuminated_sediment
      • Arena.red_FOV
      • Arena.uv_FOV
      • Arena.get_global_luminance()
      • Arena.get_global_sediment()
      • Arena.get_masked_sediment()
      • Arena.get_uv_luminance_mask()
      • Arena.reset()
    • FieldOfView
      • FieldOfView.max_visual_distance
      • FieldOfView.env_width
      • FieldOfView.env_height
      • FieldOfView.light_decay_rate
      • FieldOfView.local_scatter
      • FieldOfView.full_fov_top
      • FieldOfView.full_fov_bottom
      • FieldOfView.full_fov_left
      • FieldOfView.full_fov_right
      • FieldOfView.local_fov_top
      • FieldOfView.local_fov_bottom
      • FieldOfView.local_fov_left
      • FieldOfView.local_fov_right
      • FieldOfView.enclosed_fov_top
      • FieldOfView.enclosed_fov_bottom
      • FieldOfView.enclosed_fov_left
      • FieldOfView.enclosed_fov_right
      • FieldOfView._get_local_scatter()
      • FieldOfView.update_field_of_view()
      • FieldOfView.get_sliced_masked_image()
      • FieldOfView.get_sliced_masked_image()
      • FieldOfView.update_field_of_view()
  • simulation.constants module
  • simulation.define_actions module
    • Actions
      • Actions.add_null_action()
      • Actions.display_actions()
      • Actions.from_hdf5()
      • Actions.get_action()
      • Actions.get_all_actions()
      • Actions.get_extracted_actions()
      • Actions.get_opposing_dict()
      • Actions.sharpen_distributions()
      • Actions.to_hdf5()
    • extract_bout_sample()
    • get_angles_and_distances()
  • simulation.eye module
    • Eye
      • Eye.add_noise_to_readings()
      • Eye.read()
  • simulation.fish module
    • Fish
      • Fish.env_variables
      • Fish.body
      • Fish.rng
      • Fish.actions
      • Fish.num_actions
      • Fish.distance_to_impulse_factor
      • Fish.mouth
      • Fish.head
      • Fish.tail
      • Fish.verg_angle
      • Fish.retinal_field
      • Fish.conv_state
      • Fish.left_eye
      • Fish.right_eye
      • Fish.hungry
      • Fish.prey_consumed
      • Fish.touched_edge
      • Fish.touched_predator
      • Fish.making_capture
      • Fish.capture_possible
      • Fish.prev_action_distance
      • Fish.prev_action_angle
      • Fish.prev_action
      • Fish.energy_level
      • Fish.d_scaling_energy_cost
      • Fish.a_scaling_energy_cost
      • Fish.baseline_energy_use
      • Fish.action_energy_reward_scaling
      • Fish.touched_edge_this_step
      • Fish.deterministic_action
      • Fish.draw_angle_dist()
      • Fish.take_action()
      • Fish.update_energy_level()
  • simulation.geometry module
  • simulation.setup module
  • simulation.simfish_env module
    • BaseEnvironment
      • BaseEnvironment.action_spec()
      • BaseEnvironment.get_info()
      • BaseEnvironment.get_observation()
      • BaseEnvironment.get_predator_angles_distance()
      • BaseEnvironment.no_collision()
      • BaseEnvironment.observation_spec()
      • BaseEnvironment.reset()
      • BaseEnvironment.resolve_visual_input()
      • BaseEnvironment.step()
      • BaseEnvironment.touch_predator()
  • Module contents
• run_r2d2_simfish module
  • SimfishR2D2Builder
    • SimfishR2D2Builder.make_learner()
  • SimfishR2D2Config
    • SimfishR2D2Config.actions
  • build_experiment_config()
  • main()
• evaluate_r2d2_simfish module