Code
Hub
Workspaces
Following
Trending
Connect
MCP
copy
Create free account
hub
/
github.com/abalakrishna123/recovery-rl
/ functions
Functions
314 in github.com/abalakrishna123/recovery-rl
⨍
Functions
314
◇
Types & classes
47
↓ 87 callers
Method
to
(self, device)
recovery_rl/model.py:340
↓ 24 callers
Function
get_affine_params
(ensemble_size, in_features, out_features)
config/utils.py:15
↓ 18 callers
Function
swish
(x)
config/utils.py:6
↓ 16 callers
Method
sample
(self, state)
recovery_rl/model.py:325
↓ 14 callers
Function
get_required_argument
(dotmap, key, message, default=None)
recovery_rl/utils.py:72
↓ 12 callers
Method
__init__
(self, num_inputs, num_actions, hidden_dim)
recovery_rl/model.py:50
↓ 12 callers
Method
plot
Interface for visualizing Q_risk for all navigation environments.
recovery_rl/qrisk.py:229
↓ 12 callers
Method
step
Applies the action and steps simulation :param action: action at time-step :return: obs dict where: -each k
env/base_env.py:13
↓ 10 callers
Method
render
Renders the enviornment. Implements custom rendering support. If mode is: - dual: renders both left and main cameras - left:
env/base_env.py:64
↓ 10 callers
Method
reset
Resets this controller (clears previous solution, calls all update functions). Returns: None
recovery_rl/MPC.py:311
↓ 9 callers
Function
get_color
(algname, alt_color_map={})
plotting/plotting_utils.py:26
↓ 8 callers
Method
_get_obs
(self, images=False)
env/maze.py:170
↓ 7 callers
Method
push
(self, state, action, reward, next_state, done)
recovery_rl/replay_memory.py:21
↓ 6 callers
Method
_next_state
(self, s, a, override=False)
env/navigation2.py:98
↓ 6 callers
Method
step_cost
(self, s, a)
env/navigation2.py:105
↓ 5 callers
Function
get_legend_name
(algname, alt_name_map={})
plotting/plotting_utils.py:35
↓ 5 callers
Method
get_value
Arguments: states, actions --> list of states and list of corresponding actions to get Q_risk values for
recovery_rl/qrisk.py:184
↓ 5 callers
Method
seed
(self, seed=None)
env/base_env.py:90
↓ 5 callers
Method
update_parameters
Train task policy and associated Q function with experience in replay buffer (memory)
recovery_rl/sac.py:170
↓ 4 callers
Method
_get_obs
(self, images=False)
env/image_maze.py:200
↓ 4 callers
Method
forward
(self, inputs, ret_logvar=False)
config/maze.py:71
↓ 4 callers
Function
get_stats
(data)
plotting/plot_runs.py:14
↓ 4 callers
Function
process_obs
(obs)
env/image_maze.py:29
↓ 4 callers
Function
process_obs
(obs)
env/maze.py:29
↓ 4 callers
Function
process_obs
(obs, env_name)
recovery_rl/experiment.py:35
↓ 4 callers
Function
torchify
(x)
recovery_rl/experiment.py:26
↓ 3 callers
Method
_reset_sim
Creates a MjSim from passed in model_path :param model_path: Absolute path to model file :return: None
env/base_mujoco_env.py:37
↓ 3 callers
Method
get_distance_score
(self)
env/image_maze.py:246
↓ 3 callers
Method
get_distance_score
:return: mean of the distances between all objects and goals
env/maze.py:215
↓ 3 callers
Method
sample
(self, batch_size)
recovery_rl/replay_memory.py:27
↓ 3 callers
Method
select_action
Get action from current task policy
recovery_rl/sac.py:133
↓ 3 callers
Method
targ_proc
(obs, next_obs)
config/maze.py:138
↓ 3 callers
Method
train
Trains the internal model of this controller. Once trained, this controller switches from applying random actions to using MPC. Argum
recovery_rl/MPC.py:213
↓ 2 callers
Method
_expand_to_ts_format
(self, mat)
recovery_rl/MPC.py:441
↓ 2 callers
Method
_next_state
(self, s, a, override=False)
env/navigation1.py:99
↓ 2 callers
Function
clip_target_qpos
(target, lb, ub)
env/obj_dynamic_extraction.py:31
↓ 2 callers
Method
get_action
(self, state, train=True)
recovery_rl/experiment.py:546
↓ 2 callers
Function
hard_update
(target, source)
recovery_rl/utils.py:52
↓ 2 callers
Function
linear_schedule
(startval, endval, endtime)
recovery_rl/utils.py:62
↓ 2 callers
Function
make_env
(env_name)
env/make_utils.py:29
↓ 2 callers
Function
no_rot_dynamics
(prev_target_qpos, action)
env/obj_dynamic_extraction.py:24
↓ 2 callers
Method
project_point
(self, point, camera)
env/base_mujoco_env.py:68
↓ 2 callers
Function
register_env
(env_name)
env/make_utils.py:23
↓ 2 callers
Method
render
(self)
env/obj_dynamic_extraction.py:67
↓ 2 callers
Method
render
(self)
env/obj_extraction.py:64
↓ 2 callers
Method
reset
(self, difficulty='m', check_constraint=True, pos=())
env/image_maze.py:221
↓ 2 callers
Method
reset
(self, difficulty='h', check_constraint=True, pos=())
env/maze.py:184
↓ 2 callers
Method
sample
samples a random action from the action space.
env/navigation1.py:115
↓ 2 callers
Function
soft_update
(target, source, tau)
recovery_rl/utils.py:46
↓ 2 callers
Method
step
(self, action)
env/image_maze.py:169
↓ 2 callers
Method
step
(self, action)
env/maze.py:139
↓ 2 callers
Method
step_cost
(self, s, a)
env/navigation1.py:106
↓ 2 callers
Method
train_MB_recovery
(self, states, actions, next_states=None, epochs=50)
recovery_rl/experiment.py:251
↓ 1 callers
Method
__init__
(self, *args, **kwargs)
recovery_rl/optimizers.py:15
↓ 1 callers
Function
_create_ctrl_config
Creates controller configuration.
config/default.py:64
↓ 1 callers
Function
_create_exp_config
(exp_cfg, cfg_module, logdir, type_map)
config/default.py:54
↓ 1 callers
Method
_flatten_to_matrix
(self, ts_fmt_arr)
recovery_rl/MPC.py:457
↓ 1 callers
Method
_next_qpos
(self, action)
env/obj_dynamic_extraction.py:272
↓ 1 callers
Method
_next_qpos
(self, action)
env/obj_extraction.py:227
↓ 1 callers
Method
_next_qpos_dynamic_obs
(self, action)
env/obj_dynamic_extraction.py:278
↓ 1 callers
Method
_predict_next_obs
(self, obs, acs)
recovery_rl/MPC.py:421
↓ 1 callers
Method
act
Returns the action that this controller would take at time t given observation obs. Arguments: obs: The current observation
recovery_rl/MPC.py:322
↓ 1 callers
Method
agent_setup
(self, env)
recovery_rl/experiment.py:169
↓ 1 callers
Function
clip_target_qpos
(target, lb, ub)
env/obj_extraction.py:29
↓ 1 callers
Method
compute_decays
(self)
config/maze.py:52
↓ 1 callers
Function
create_conditional
(cl, cond, message)
config/default.py:136
↓ 1 callers
Function
create_config
(env_name, ctrl_type, ctrl_args, overrides, logdir)
config/default.py:15
↓ 1 callers
Method
dump_logs
(self, train_rollouts, test_rollouts)
recovery_rl/experiment.py:540
↓ 1 callers
Method
experiment_setup
(self)
recovery_rl/experiment.py:89
↓ 1 callers
Method
expert_action
(self)
env/image_maze.py:250
↓ 1 callers
Method
expert_action
(self)
env/maze.py:222
↓ 1 callers
Method
fit_input_stats
(self, data)
config/maze.py:61
↓ 1 callers
Method
forward
(self, state)
recovery_rl/model.py:317
↓ 1 callers
Method
forward
(self, state)
recovery_rl/model.py:407
↓ 1 callers
Method
forward
(self, state)
recovery_rl/model.py:468
↓ 1 callers
Method
forward
(self, state)
recovery_rl/model.py:512
↓ 1 callers
Method
forward
(self, state)
recovery_rl/model.py:594
↓ 1 callers
Method
forward
(self, state)
recovery_rl/model.py:680
↓ 1 callers
Function
get_args
()
arg_utils.py:8
↓ 1 callers
Method
get_contact_info
(self)
env/obj_dynamic_extraction.py:154
↓ 1 callers
Method
get_distance_score
:return: mean of the distances between all objects and goals
env/base_mujoco_env.py:140
↓ 1 callers
Function
get_img_from_fig
(fig, dpi=180)
env/navigation1.py:168
↓ 1 callers
Function
get_img_from_fig
(fig, dpi=180)
env/navigation2.py:247
↓ 1 callers
Method
get_offline_data
(self)
recovery_rl/experiment.py:177
↓ 1 callers
Method
get_test_rollout
(self, i_episode)
recovery_rl/experiment.py:493
↓ 1 callers
Method
get_train_rollout
(self, i_episode)
recovery_rl/experiment.py:379
↓ 1 callers
Function
moving_average
(x, N)
plotting/plot_runs.py:23
↓ 1 callers
Function
no_rot_dynamics
(prev_target_qpos, action)
env/obj_extraction.py:22
↓ 1 callers
Function
npy_to_gif
(im_list, filename, fps=4)
recovery_rl/experiment.py:29
↓ 1 callers
Function
npy_to_gif
(im_list, filename, fps=4)
recovery_rl/VisualMPC.py:60
↓ 1 callers
Method
object_reset_poses
(self)
env/obj_dynamic_extraction.py:232
↓ 1 callers
Method
object_reset_poses
(self)
env/obj_extraction.py:187
↓ 1 callers
Method
obs_postproc
(obs, pred)
config/maze.py:134
↓ 1 callers
Method
obtain_solution
(self, *args, **kwargs)
recovery_rl/optimizers.py:24
↓ 1 callers
Function
plot_experiment
Construct experiment map for this experiment
plotting/plot_runs.py:141
↓ 1 callers
Method
pretrain_critic_recovery
(self)
recovery_rl/experiment.py:261
↓ 1 callers
Method
pretrain_task_critic
(self)
recovery_rl/experiment.py:337
↓ 1 callers
Function
process_action
(a)
env/image_maze.py:25
↓ 1 callers
Function
process_action
(a)
env/navigation1.py:50
↓ 1 callers
Function
process_action
(a)
env/maze.py:25
next →
1–100 of 314, ranked by callers