SYMBOL INDEX (224 symbols across 31 files)

FILE: robot_nav/SIM_ENV/marl_sim.py
  class MARL_SIM (line 9) | class MARL_SIM(SIM_ENV):
    method __init__ (line 24) | def __init__(
    method step (line 55) | def step(self, action, connection, combined_weights=None):
    method reset (line 207) | def reset(
    method get_reward (line 314) | def get_reward(goal, collision, action, closest_robots, distance, phas...

FILE: robot_nav/SIM_ENV/sim.py
  class SIM (line 8) | class SIM(SIM_ENV):
    method __init__ (line 20) | def __init__(self, world_file="robot_world.yaml", disable_plotting=Fal...
    method step (line 35) | def step(self, lin_velocity=0.0, ang_velocity=0.1):
    method reset (line 68) | def reset(
    method get_reward (line 124) | def get_reward(goal, collision, action, laser_scan):

FILE: robot_nav/SIM_ENV/sim_env.py
  class SIM_ENV (line 5) | class SIM_ENV(ABC):
    method step (line 7) | def step(self):
    method reset (line 11) | def reset(self):
    method cossin (line 15) | def cossin(vec1, vec2):
    method get_reward (line 34) | def get_reward():

FILE: robot_nav/marl_test_random.py
  function outside_of_bounds (line 13) | def outside_of_bounds(poses):
  function main (line 33) | def main(args=None):

FILE: robot_nav/marl_test_single.py
  class SINGLE_SIM (line 14) | class SINGLE_SIM(SIM_ENV):
    method __init__ (line 29) | def __init__(
    method step (line 51) | def step(self, action, connection, combined_weights=None):
    method reset (line 177) | def reset(
    method get_reward (line 231) | def get_reward(goal, collision, action, closest_robots, distance, phas...
  function outside_of_bounds (line 279) | def outside_of_bounds(poses):
  function main (line 299) | def main(args=None):

FILE: robot_nav/marl_train.py
  function outside_of_bounds (line 11) | def outside_of_bounds(poses):
  function main (line 31) | def main(args=None):

FILE: robot_nav/models/CNNTD3/CNNTD3.py
  class Actor (line 13) | class Actor(nn.Module):
    method __init__ (line 32) | def __init__(self, action_dim):
    method forward (line 49) | def forward(self, s):
  class Critic (line 85) | class Critic(nn.Module):
    method __init__ (line 104) | def __init__(self, action_dim):
    method forward (line 131) | def forward(self, s, action):
  class CNNTD3 (line 180) | class CNNTD3(object):
    method __init__ (line 205) | def __init__(
    method get_action (line 246) | def get_action(self, obs, add_noise):
    method act (line 264) | def act(self, state):
    method train (line 279) | def train(
    method save (line 418) | def save(self, filename, directory):
    method load (line 438) | def load(self, filename, directory):
    method prepare_state (line 460) | def prepare_state(self, latest_scan, distance, cos, sin, collision, go...

FILE: robot_nav/models/DDPG/DDPG.py
  class Actor (line 13) | class Actor(nn.Module):
    method __init__ (line 31) | def __init__(self, state_dim, action_dim):
    method forward (line 41) | def forward(self, s):
  class Critic (line 57) | class Critic(nn.Module):
    method __init__ (line 77) | def __init__(self, state_dim, action_dim):
    method forward (line 89) | def forward(self, s, a):
  class DDPG (line 112) | class DDPG(object):
    method __init__ (line 135) | def __init__(
    method get_action (line 176) | def get_action(self, obs, add_noise):
    method act (line 194) | def act(self, state):
    method train (line 208) | def train(
    method save (line 344) | def save(self, filename, directory):
    method load (line 364) | def load(self, filename, directory):
    method prepare_state (line 386) | def prepare_state(self, latest_scan, distance, cos, sin, collision, go...

FILE: robot_nav/models/HCM/hardcoded_model.py
  class HCM (line 9) | class HCM(object):
    method __init__ (line 27) | def __init__(
    method get_action (line 53) | def get_action(self, state, add_noise):
    method train (line 88) | def train(
    method save (line 117) | def save(self, filename, directory):
    method load (line 130) | def load(self, filename, directory):
    method prepare_state (line 143) | def prepare_state(self, latest_scan, distance, cos, sin, collision, go...

FILE: robot_nav/models/MARL/Attention/g2anet.py
  class G2ANet (line 7) | class G2ANet(nn.Module):
    method __init__ (line 29) | def __init__(self, embedding_dim):
    method encode_agent_features (line 52) | def encode_agent_features(self, embed):
    method forward (line 66) | def forward(self, embedding):

FILE: robot_nav/models/MARL/Attention/iga.py
  function entropy_from_attention (line 9) | def entropy_from_attention(attn_weights, target_nodes, num_nodes, eps=1e...
  class GoalAttentionLayer (line 48) | class GoalAttentionLayer(MessagePassing):
    method __init__ (line 62) | def __init__(self, node_dim, edge_dim, out_dim):
    method forward (line 74) | def forward(self, x, edge_index, edge_attr):
    method message (line 94) | def message(self, x_i, edge_attr, index, ptr, size_i):
    method aggregate (line 116) | def aggregate(self, inputs, index, ptr=None, dim_size=None):
  class Attention (line 132) | class Attention(nn.Module):
    method __init__ (line 158) | def __init__(self, embedding_dim):
    method encode_agent_features (line 210) | def encode_agent_features(self, embed):
    method forward (line 224) | def forward(self, embedding):

FILE: robot_nav/models/MARL/marlTD3/marlTD3.py
  class Actor (line 14) | class Actor(nn.Module):
    method __init__ (line 32) | def __init__(self, action_dim, embedding_dim, attention):
    method forward (line 51) | def forward(self, obs, detach_attn=False):
  class Critic (line 79) | class Critic(nn.Module):
    method __init__ (line 97) | def __init__(self, action_dim, embedding_dim, attention):
    method forward (line 131) | def forward(self, embedding, action):
  class TD3 (line 171) | class TD3(object):
    method __init__ (line 208) | def __init__(
    method get_action (line 268) | def get_action(self, obs, add_noise):
    method act (line 290) | def act(self, state):
    method train (line 310) | def train(
    method save (line 531) | def save(self, filename, directory):
    method load (line 551) | def load(self, filename, directory):
    method prepare_state (line 573) | def prepare_state(

FILE: robot_nav/models/PPO/PPO.py
  class RolloutBuffer (line 10) | class RolloutBuffer:
    method __init__ (line 23) | def __init__(self):
    method clear (line 34) | def clear(self):
    method add (line 45) | def add(self, state, action, reward, terminal, next_state):
  class ActorCritic (line 61) | class ActorCritic(nn.Module):
    method __init__ (line 73) | def __init__(self, state_dim, action_dim, action_std_init, max_action,...
    method set_action_std (line 111) | def set_action_std(self, new_action_std):
    method forward (line 122) | def forward(self):
    method act (line 131) | def act(self, state, sample):
    method evaluate (line 157) | def evaluate(self, state, action):
  class PPO (line 185) | class PPO:
    method __init__ (line 210) | def __init__(
    method set_action_std (line 264) | def set_action_std(self, new_action_std):
    method decay_action_std (line 275) | def decay_action_std(self, action_std_decay_rate, min_action_std):
    method get_action (line 300) | def get_action(self, state, add_noise):
    method train (line 324) | def train(self, replay_buffer, iterations, batch_size):
    method prepare_state (line 423) | def prepare_state(self, latest_scan, distance, cos, sin, collision, go...
    method save (line 468) | def save(self, filename, directory):
    method load (line 481) | def load(self, filename, directory):

FILE: robot_nav/models/RCPG/RCPG.py
  class Actor (line 11) | class Actor(nn.Module):
    method __init__ (line 29) | def __init__(self, action_dim, rnn="gru"):
    method forward (line 60) | def forward(self, s):
  class Critic (line 92) | class Critic(nn.Module):
    method __init__ (line 109) | def __init__(self, action_dim, rnn="gru"):
    method forward (line 151) | def forward(self, s, action):
  class RCPG (line 194) | class RCPG(object):
    method __init__ (line 227) | def __init__(
    method get_action (line 265) | def get_action(self, obs, add_noise):
    method act (line 288) | def act(self, state):
    method train (line 307) | def train(
    method save (line 426) | def save(self, filename, directory):
    method load (line 449) | def load(self, filename, directory):
    method prepare_state (line 474) | def prepare_state(self, latest_scan, distance, cos, sin, collision, go...

FILE: robot_nav/models/SAC/SAC.py
  class SAC (line 13) | class SAC(object):
    method __init__ (line 45) | def __init__(
    method save (line 146) | def save(self, filename, directory):
    method load (line 162) | def load(self, filename, directory):
    method train (line 181) | def train(self, replay_buffer, iterations, batch_size):
    method alpha (line 205) | def alpha(self):
    method get_action (line 212) | def get_action(self, obs, add_noise):
    method act (line 230) | def act(self, obs, sample=False):
    method update_critic (line 249) | def update_critic(self, obs, action, reward, next_obs, done, step):
    method update_actor_and_alpha (line 284) | def update_actor_and_alpha(self, obs, step):
    method update (line 329) | def update(self, replay_buffer, step, batch_size):
    method prepare_state (line 364) | def prepare_state(self, latest_scan, distance, cos, sin, collision, go...

FILE: robot_nav/models/SAC/SAC_actor.py
  class TanhTransform (line 10) | class TanhTransform(pyd.transforms.Transform):
    method __init__ (line 29) | def __init__(self, cache_size=1):
    method atanh (line 39) | def atanh(x):
    method __eq__ (line 51) | def __eq__(self, other):
    method _call (line 60) | def _call(self, x):
    method _inverse (line 72) | def _inverse(self, y):
    method log_abs_det_jacobian (line 86) | def log_abs_det_jacobian(self, x, y):
  class SquashedNormal (line 102) | class SquashedNormal(pyd.transformed_distribution.TransformedDistribution):
    method __init__ (line 109) | def __init__(self, loc, scale):
    method mean (line 125) | def mean(self):
  class DiagGaussianActor (line 138) | class DiagGaussianActor(nn.Module):
    method __init__ (line 153) | def __init__(self, obs_dim, action_dim, hidden_dim, hidden_depth, log_...
    method forward (line 165) | def forward(self, obs):
    method log (line 190) | def log(self, writer, step):

FILE: robot_nav/models/SAC/SAC_critic.py
  class DoubleQCritic (line 7) | class DoubleQCritic(nn.Module):
    method __init__ (line 21) | def __init__(self, obs_dim, action_dim, hidden_dim, hidden_depth):
    method forward (line 34) | def forward(self, obs, action):
    method log (line 56) | def log(self, writer, step):

FILE: robot_nav/models/SAC/SAC_utils.py
  function soft_update_params (line 12) | def soft_update_params(net, target_net, tau):
  function set_seed_everywhere (line 26) | def set_seed_everywhere(seed):
  function make_dir (line 40) | def make_dir(*path_parts):
  function weight_init (line 58) | def weight_init(m):
  class MLP (line 73) | class MLP(nn.Module):
    method __init__ (line 85) | def __init__(
    method forward (line 92) | def forward(self, x):
  function mlp (line 105) | def mlp(input_dim, hidden_dim, output_dim, hidden_depth, output_mod=None):
  function to_np (line 132) | def to_np(t):

FILE: robot_nav/models/TD3/TD3.py
  class Actor (line 13) | class Actor(nn.Module):
    method __init__ (line 31) | def __init__(self, state_dim, action_dim):
    method forward (line 41) | def forward(self, s):
  class Critic (line 57) | class Critic(nn.Module):
    method __init__ (line 75) | def __init__(self, state_dim, action_dim):
    method forward (line 96) | def forward(self, s, a):
  class TD3 (line 128) | class TD3(object):
    method __init__ (line 129) | def __init__(
    method get_action (line 190) | def get_action(self, obs, add_noise):
    method act (line 208) | def act(self, state):
    method train (line 222) | def train(
    method save (line 361) | def save(self, filename, directory):
    method load (line 381) | def load(self, filename, directory):
    method prepare_state (line 403) | def prepare_state(self, latest_scan, distance, cos, sin, collision, go...

FILE: robot_nav/replay_buffer.py
  class ReplayBuffer (line 8) | class ReplayBuffer(object):
    method __init__ (line 16) | def __init__(self, buffer_size, random_seed=123):
    method add (line 29) | def add(self, s, a, r, t, s2):
    method size (line 48) | def size(self):
    method sample_batch (line 57) | def sample_batch(self, batch_size):
    method return_buffer (line 80) | def return_buffer(self):
    method clear (line 95) | def clear(self):
  class RolloutReplayBuffer (line 103) | class RolloutReplayBuffer(object):
    method __init__ (line 110) | def __init__(self, buffer_size, random_seed=123, history_len=10):
    method add (line 126) | def add(self, s, a, r, t, s2):
    method size (line 147) | def size(self):
    method sample_batch (line 156) | def sample_batch(self, batch_size):
    method clear (line 216) | def clear(self):

FILE: robot_nav/rl_test.py
  function main (line 9) | def main(args=None):

FILE: robot_nav/rl_test_random.py
  function main (line 11) | def main(args=None):

FILE: robot_nav/rl_train.py
  function main (line 9) | def main(args=None):
  function evaluate (line 106) | def evaluate(model, epoch, sim, eval_episodes=10):

FILE: robot_nav/rnn_test.py
  function main (line 11) | def main(args=None):

FILE: robot_nav/rnn_train.py
  function main (line 12) | def main(args=None):
  function evaluate (line 126) | def evaluate(model, epoch, sim, history_len, max_steps, eval_episodes=10):

FILE: robot_nav/rvo_test_random.py
  class RVO_RANDOM_SIM (line 17) | class RVO_RANDOM_SIM(SIM_ENV):
    method __init__ (line 33) | def __init__(self, world_file="multi_robot_world.yaml", disable_plotti...
    method step (line 53) | def step(self):
    method reset (line 95) | def reset(
    method get_reward (line 180) | def get_reward(self):
  function outside_of_bounds (line 194) | def outside_of_bounds(poses):
  function main (line 216) | def main(args=None):

FILE: robot_nav/rvo_test_single.py
  class RVO_SINGLE_SIM (line 14) | class RVO_SINGLE_SIM(SIM_ENV):
    method __init__ (line 30) | def __init__(self, world_file="multi_robot_world.yaml", disable_plotti...
    method step (line 50) | def step(self):
    method reset (line 81) | def reset(
    method get_reward (line 113) | def get_reward(self):
  function outside_of_bounds (line 126) | def outside_of_bounds(poses):
  function main (line 148) | def main(args=None):

FILE: robot_nav/utils.py
  class Pretraining (line 11) | class Pretraining:
    method __init__ (line 22) | def __init__(
    method load_buffer (line 34) | def load_buffer(self):
    method train (line 88) | def train(
  function get_buffer (line 114) | def get_buffer(
  function get_max_bound (line 182) | def get_max_bound(

FILE: tests/test_model.py
  function test_models (line 29) | def test_models(model, state_dim):
  function test_max_bound_models (line 68) | def test_max_bound_models(model, state_dim):
  function test_marl_models (line 100) | def test_marl_models():

FILE: tests/test_sim.py
  function test_sim (line 16) | def test_sim():
  function test_marl_sim (line 33) | def test_marl_sim():
  function test_sincos (line 56) | def test_sincos():

FILE: tests/test_utils.py
  function test_buffer (line 12) | def test_buffer():
  function test_rollout_buffer (line 51) | def test_rollout_buffer():
  function test_ppo_buffer (line 90) | def test_ppo_buffer():