Index

A

add_neuron_group() (Agent method) (Neurons method) Agent (class in rtgym.agent.agent) AgentState (class in rtgym.dataclass.agent_state) (class in rtgym.dataclasses) aggregate_res_maps() (Neurons method) append() (RawTrajectory method) apply_gaussian_blur() (Masking method) arena (Agent property) (BaseBehavior property)

Arena (class in rtgym.arena.arena) arena (Neurons property) arena_height (Arena property) arena_map (Arena property) (RatatouGym property) arena_width (Arena property) as_numpy() (AgentState method) (Trajectory method) AutonomousBehavior (class in rtgym.agent.behavior.autonomous_behavior) avoid_boundary_dist (BehaviorConfig property)

B

BaseBehavior (class in rtgym.agent.behavior.base_behavior) batch_size (RawAgentState property) Behavior (class in rtgym.agent.behavior) (class in rtgym.agent.behavior.behavior)

BehaviorConfig (class in rtgym.agent.behavior.behavior_config) BoundaryCell (class in rtgym.agent.sensory.spatial_modulated) (class in rtgym.agent.sensory.spatial_modulated.boundary_cell)

C

clone() (AgentState method) combine_trajectories() (in module rtgym.utils.data_processing) common_params (Neurons property) compute_layout() (in module rtgym.utils.verbose) compute_res() (Neurons method) ControllableBehavior (class in rtgym.agent.behavior.controllable_behavior)

coord (AgentState property) (Trajectory property) copy() (AgentState method) (RawAgentState method) (Trajectory method), [1] create_dataclass_result() (in module rtgym.utils.decode_response)

D

decode_response() (Neurons method) decode_response_euclidean() (in module rtgym.utils.decode_response) decode_response_faiss() (in module rtgym.utils.decode_response) decode_response_interpolation() (in module rtgym.utils.decode_response) decode_response_kdtree() (in module rtgym.utils.decode_response) decode_response_torch() (in module rtgym.utils.decode_response) diffusion_smooth() (DiffusionCell method), [1] DiffusionCell (class in rtgym.agent.sensory.spatial_modulated) (class in rtgym.agent.sensory.spatial_modulated.diffusion_cell)

DirectionCell (class in rtgym.agent.sensory.movement_modulated) (class in rtgym.agent.sensory.movement_modulated.direction_cell) DirectionRad (class in rtgym.agent.sensory.movement_modulated) (class in rtgym.agent.sensory.movement_modulated.direction_rad) disp (AgentState property) (Trajectory property) DisplacementAbs (class in rtgym.agent.sensory.movement_modulated) (class in rtgym.agent.sensory.movement_modulated.displacement_abs)

F

fields (AgentState attribute) (Trajectory attribute) filter_neurons() (Neurons method) float_coord (AgentState property) (RawAgentState property) (Trajectory property)

free_space (Arena property) from_dict() (Trajectory class method), [1] from_numpy() (Trajectory class method)

G

gaussian_kernel_1d() (Masking method) generate_box_arena() (in module rtgym.arena.arena_shapes.box) generate_carpenter_rooms_arena() (in module rtgym.arena.arena_shapes.carpenter_rooms) generate_circle_arena() (in module rtgym.arena.arena_shapes.circle) generate_cornered_rectangle_arena() (in module rtgym.arena.arena_shapes.cornered_rectangle) generate_hairpin_arena() (in module rtgym.arena.arena_shapes.hairpin) generate_maze_0_arena() (in module rtgym.arena.arena_shapes.maze_0) generate_maze_1_arena() (in module rtgym.arena.arena_shapes.maze_1) generate_maze_2_arena() (in module rtgym.arena.arena_shapes.maze_2) generate_random_pos() (Arena method) generate_random_pos_numpy() (Arena method) generate_rectangle_arena() (in module rtgym.arena.arena_shapes.rectangle) generate_titles() (in module rtgym.utils.verbose) generate_trainer_0_arena() (in module rtgym.arena.arena_shapes.trainer_0) generate_trajectory() (AutonomousBehavior method) (Behavior method), [1] (TrajectoryGenerator method) generate_triangle_arena() (in module rtgym.arena.arena_shapes.triangle) generate_two_rooms_arena() (in module rtgym.arena.arena_shapes.two_rooms) get_gym_dimensions() (in module rtgym.utils.data_processing) get_ratemap() (RatemapAggregator method) get_response() (Agent method) (DirectionCell method), [1] (DisplacementAbs method), [1] (HeadDirectionCell method), [1] (Neurons method) (SMBase method) (SpeedCell method), [1]

get_responses() (DirectionRad method), [1] get_specs() (Arena method) (BoundaryCell method), [1] (DirectionCell method), [1] (DirectionRad method), [1] (DisplacementAbs method), [1] (GridCell method), [1] (HeadDirectionCell method), [1] (MMBase method) (PlaceCell method), [1] (SMBase method) (SpeedCell method), [1] (WeakSMCell method), [1] get_trajectory() (ControllableBehavior method) GridCell (class in rtgym.agent.sensory.spatial_modulated) (class in rtgym.agent.sensory.spatial_modulated.grid_cell)

H

hairpin_waypoints() (in module rtgym.arena.arena_shapes.hairpin) hash_seed() (in module rtgym.utils.common) hd (AgentState property) (Trajectory property)

HeadDirectionCell (class in rtgym.agent.sensory.movement_modulated) (class in rtgym.agent.sensory.movement_modulated.head_direction_cell)

I

init_arena_map() (Arena method) (RatatouGym method) init_control() (Agent method) init_counts() (RatemapAggregator method) init_from_profile() (AutonomousBehavior method) (Behavior method), [1] (ControllableBehavior method) (Neurons method) (TrajectoryGenerator method)

init_neurons() (Agent method) initialized (BaseBehavior attribute) int_coord (AgentState property), [1] (RawAgentState property) (Trajectory property), [1] inv_arena_map (Arena property) (RatatouGym property) invmap_ (Arena property)

L

list_all() (Neurons method) load() (Arena method) (Neurons method) (SMBase class method) (Trajectory static method), [1]

load_arena() (RatatouGym method) load_from_dict() (GridCell class method), [1] load_from_state_dict() (Neurons method)

M

map (Arena property) map_ (Arena property) mask() (Masking method) Masking (class in rtgym.utils.masking) MMBase (class in rtgym.agent.sensory.movement_modulated.mm_base) module rtgym rtgym.agent rtgym.agent.agent rtgym.agent.behavior rtgym.agent.behavior.autonomous_behavior rtgym.agent.behavior.base_behavior rtgym.agent.behavior.behavior rtgym.agent.behavior.behavior_config rtgym.agent.behavior.controllable_behavior rtgym.agent.control rtgym.agent.neurons rtgym.agent.sensory rtgym.agent.sensory.movement_modulated rtgym.agent.sensory.movement_modulated.direction_cell rtgym.agent.sensory.movement_modulated.direction_rad rtgym.agent.sensory.movement_modulated.displacement_abs rtgym.agent.sensory.movement_modulated.head_direction_cell rtgym.agent.sensory.movement_modulated.mm_base rtgym.agent.sensory.movement_modulated.speed_cell rtgym.agent.sensory.spatial_modulated rtgym.agent.sensory.spatial_modulated.boundary_cell rtgym.agent.sensory.spatial_modulated.diffusion_cell rtgym.agent.sensory.spatial_modulated.grid_cell rtgym.agent.sensory.spatial_modulated.place_cell rtgym.agent.sensory.spatial_modulated.sm_base rtgym.agent.sensory.spatial_modulated.weak_sm_cell rtgym.arena rtgym.arena.arena rtgym.arena.arena_shapes rtgym.arena.arena_shapes.box rtgym.arena.arena_shapes.carpenter_rooms rtgym.arena.arena_shapes.circle rtgym.arena.arena_shapes.cornered_rectangle rtgym.arena.arena_shapes.hairpin rtgym.arena.arena_shapes.maze_0 rtgym.arena.arena_shapes.maze_1 rtgym.arena.arena_shapes.maze_2 rtgym.arena.arena_shapes.rectangle rtgym.arena.arena_shapes.trainer_0 rtgym.arena.arena_shapes.triangle rtgym.arena.arena_shapes.two_rooms rtgym.dataclass rtgym.dataclass.agent_state rtgym.dataclass.raw_agent_state rtgym.dataclass.raw_trajectory rtgym.dataclass.trajectory rtgym.dataclasses rtgym.rtgym rtgym.utils rtgym.utils.common rtgym.utils.data_processing rtgym.utils.decode_response rtgym.utils.masking rtgym.utils.verbose rtgym.utils.visualization rtgym.utils.visualization.plot_ratemap

N

n_steps (Trajectory property), [1] ndim (Arena property) neuron_groups (Agent property) Neurons (class in rtgym.agent.neurons)

new_mask() (Masking method) notify_subscribers() (Arena method) num_neurons() (Agent method) (Neurons method)

P

PlaceCell (class in rtgym.agent.sensory.spatial_modulated) (class in rtgym.agent.sensory.spatial_modulated.place_cell) plot_dual_field() (in module rtgym.utils.verbose) plot_dual_fields() (in module rtgym.utils.verbose) plot_ratemaps() (in module rtgym.utils.visualization.plot_ratemap) plot_single_field() (in module rtgym.utils.verbose)

point_in_triangle() (in module rtgym.arena.arena_shapes.triangle) prepare_fields() (in module rtgym.utils.verbose) print_dict() (in module rtgym.utils.verbose) print_specs() (Arena method) (MMBase method) (SMBase method)

R

random_pos() (RatatouGym method) random_spawn() (Agent method) random_traverse() (Agent method) RatatouGym (class in rtgym.rtgym) RatemapAggregator (class in rtgym.utils.data_processing) RawAgentState (class in rtgym.dataclass.raw_agent_state) RawTrajectory (class in rtgym.dataclass.raw_trajectory) register_callback() (BehaviorConfig method) require_init() (BaseBehavior static method) reset() (AgentState method) (Behavior method), [1] (ControllableBehavior method) (RatemapAggregator method) (RawAgentState method) (TrajectoryGenerator method) reshape() (Trajectory method), [1] restrain2ff() (in module rtgym.utils.data_processing) rtgym module rtgym.agent module rtgym.agent.agent module rtgym.agent.behavior module rtgym.agent.behavior.autonomous_behavior module rtgym.agent.behavior.base_behavior module rtgym.agent.behavior.behavior module rtgym.agent.behavior.behavior_config module rtgym.agent.behavior.controllable_behavior module rtgym.agent.control module rtgym.agent.neurons module rtgym.agent.sensory module rtgym.agent.sensory.movement_modulated module rtgym.agent.sensory.movement_modulated.direction_cell module rtgym.agent.sensory.movement_modulated.direction_rad module rtgym.agent.sensory.movement_modulated.displacement_abs module rtgym.agent.sensory.movement_modulated.head_direction_cell module rtgym.agent.sensory.movement_modulated.mm_base module rtgym.agent.sensory.movement_modulated.speed_cell module rtgym.agent.sensory.spatial_modulated module rtgym.agent.sensory.spatial_modulated.boundary_cell module rtgym.agent.sensory.spatial_modulated.diffusion_cell module rtgym.agent.sensory.spatial_modulated.grid_cell module rtgym.agent.sensory.spatial_modulated.place_cell module

rtgym.agent.sensory.spatial_modulated.sm_base module rtgym.agent.sensory.spatial_modulated.weak_sm_cell module rtgym.arena module rtgym.arena.arena module rtgym.arena.arena_shapes module rtgym.arena.arena_shapes.box module rtgym.arena.arena_shapes.carpenter_rooms module rtgym.arena.arena_shapes.circle module rtgym.arena.arena_shapes.cornered_rectangle module rtgym.arena.arena_shapes.hairpin module rtgym.arena.arena_shapes.maze_0 module rtgym.arena.arena_shapes.maze_1 module rtgym.arena.arena_shapes.maze_2 module rtgym.arena.arena_shapes.rectangle module rtgym.arena.arena_shapes.trainer_0 module rtgym.arena.arena_shapes.triangle module rtgym.arena.arena_shapes.two_rooms module rtgym.dataclass module rtgym.dataclass.agent_state module rtgym.dataclass.raw_agent_state module rtgym.dataclass.raw_trajectory module rtgym.dataclass.trajectory module rtgym.dataclasses module rtgym.rtgym module rtgym.utils module rtgym.utils.common module rtgym.utils.data_processing module rtgym.utils.decode_response module rtgym.utils.masking module rtgym.utils.verbose module rtgym.utils.visualization module rtgym.utils.visualization.plot_ratemap module

S

s_res (BaseBehavior property) (Neurons property) (RatatouGym property) save() (Arena method) (Neurons method) (SMBase method) save_arena() (RatatouGym method) sens_category (SMBase attribute) sens_type (BoundaryCell attribute), [1] (DiffusionCell attribute), [1] (DirectionCell attribute), [1] (DirectionRad attribute), [1] (DisplacementAbs attribute), [1] (GridCell attribute), [1] (HeadDirectionCell attribute), [1] (MMBase attribute) (PlaceCell attribute), [1] (SMBase attribute) (SpeedCell attribute), [1] (WeakSMCell attribute), [1] set_arena_map() (Arena method) (RatatouGym method)

set_m_max() (Masking method) set_m_min() (Masking method) set_sensory_manually() (RatatouGym method) size (Trajectory property) slice() (Trajectory method) SMBase (class in rtgym.agent.sensory.spatial_modulated.sm_base) spawn() (Agent method) SpeedCell (class in rtgym.agent.sensory.movement_modulated) (class in rtgym.agent.sensory.movement_modulated.speed_cell) state (Agent property) state_dict() (GridCell method), [1] (Trajectory method), [1] states2ff() (in module rtgym.utils.data_processing) states_to_ratemap() (in module rtgym.utils.data_processing) step() (Agent method) (Behavior method), [1] (ControllableBehavior method) (TrajectoryGenerator method) step_state() (Agent method) subscribe() (Arena method)

T

t_range() (Trajectory method), [1] t_res (BaseBehavior property) (Neurons property) (RatatouGym property) tensor_map() (Arena method) title_single_field() (in module rtgym.utils.verbose) to() (AgentState method) (Masking method) (Neurons method) (SMBase method) (Trajectory method)

to_agent_state() (RawAgentState method) to_coord() (RatatouGym method) to_numpy() (AgentState method) to_sec() (RatatouGym method) to_tensor() (Masking method) to_trajectory() (RawTrajectory method) to_ts() (RatatouGym method) Trajectory (class in rtgym.dataclass.trajectory) (class in rtgym.dataclasses) TrajectoryGenerator (class in rtgym.agent.control)

U

update() (RatemapAggregator method)

update_state() (RawTrajectory method)

V

validate_index() (Arena method) vis() (Arena method) (DirectionCell method), [1] (HeadDirectionCell method), [1] (MMBase method) (SMBase method) (SpeedCell method), [1]

vis_gif() (RatatouGym method) vis_traj() (RatatouGym method) visualize_fields() (in module rtgym.utils.verbose)

W

WeakSMCell (class in rtgym.agent.sensory.spatial_modulated) (class in rtgym.agent.sensory.spatial_modulated.weak_sm_cell)