MCPcopy Create free account

hub / github.com/TJU-DRL-LAB/AI-Optimizer / functions

Functions3,230 in github.com/TJU-DRL-LAB/AI-Optimizer

↓ 1 callersMethodclose
Properly close the game.
modelbased-rl/SampledMuZero/games/atari.py:187
↓ 1 callersMethodclose
Properly close the game.
modelbased-rl/SampledMuZero/games/lunarlander.py:172
↓ 1 callersMethodclose
(self)
multiagent-rl/easy-marl/envs/continuous_mpe/multiagent/rendering.py:340
↓ 1 callersMethodclose_game
(self)
modelbased-rl/SampledMuZero/self_play.py:215
↓ 1 callersFunctioncollect
Collects data via interaction with environment. Args: algo: algorithm object. env: gym-like environment. buffer : replay
offline-rl-algorithms/E2O/d3rlpy_new/d3rlpy/online/iterators.py:464
↓ 1 callersMethodcollect
Collects data via interaction with environment. If ``buffer`` is not given, ``ReplayBuffer`` will be internally created. Args:
offline-rl-algorithms/E2O/d3rlpy_new/d3rlpy/algos/base.py:350
↓ 1 callersFunctioncollect_rollouts
( step, env_ctor, duration, num_agents, agent_config, isolate_envs)
modelbased-rl/PlaNet/planet/control/simulate.py:61
↓ 1 callersMethodcompute_actor_loss
(self, batch: TorchMiniBatch)
offline-rl-algorithms/E2O/d3rlpy_new/d3rlpy/algos/torch/sac_impl.py:387
↓ 1 callersMethodcompute_actor_loss
(self, batch: TorchMiniBatch)
offline-rl-algorithms/E2O/d3rlpy_new/d3rlpy/algos/torch/ddpg_impl.py:186
↓ 1 callersMethodcompute_actor_loss
(self, batch: TorchMiniBatch)
offline-rl-algorithms/E2O/d3rlpy_new/d3rlpy/algos/torch/awac_impl.py:103
↓ 1 callersMethodcompute_actor_loss
(self, batch: TorchMiniBatch)
offline-rl-algorithms/REDQ/redq_impl.py:358
↓ 1 callersMethodcompute_actor_loss
(self, batch: TorchMiniBatch)
offline-rl-algorithms/COMBO/t.py:358
↓ 1 callersMethodcompute_critic_loss
( self, batch: TorchMiniBatch, q_tpn: torch.Tensor, )
offline-rl-algorithms/E2O/d3rlpy_new/d3rlpy/algos/torch/sac_impl.py:354
↓ 1 callersMethodcompute_critic_loss
( self, batch: TorchMiniBatch, q_tpn: torch.Tensor )
offline-rl-algorithms/E2O/d3rlpy_new/d3rlpy/algos/torch/cql_impl.py:110
↓ 1 callersMethodcompute_critic_loss
( self, batch: TorchMiniBatch, q_tpn: torch.Tensor )
offline-rl-algorithms/E2O/d3rlpy_new/d3rlpy/algos/torch/ddpg_impl.py:154
↓ 1 callersMethodcompute_critic_loss
( self, batch: TorchMiniBatch, q_tpn: torch.Tensor, )
offline-rl-algorithms/REDQ/redq_impl.py:325
↓ 1 callersMethodcompute_critic_loss
( self, batch: TorchMiniBatch, q_tpn: torch.Tensor, )
offline-rl-algorithms/COMBO/t.py:325
↓ 1 callersMethodcompute_epsilon
Returns decayed :math:`\\epsilon`. Returns: :math:`\\epsilon`.
offline-rl-algorithms/E2O/d3rlpy_new/d3rlpy/online/explorers.py:99
↓ 1 callersMethodcompute_error
( self, observations: torch.Tensor, actions: torch.Tensor, rewards: torch.Tens
offline-rl-algorithms/E2O/d3rlpy_new/d3rlpy/models/torch/dynamics.py:113
↓ 1 callersMethodcompute_imitator_loss
(self, batch: TorchMiniBatch)
offline-rl-algorithms/E2O/d3rlpy_new/d3rlpy/algos/torch/bear_impl.py:210
↓ 1 callersMethodcompute_imitator_loss
(self, batch: TorchMiniBatch)
offline-rl-algorithms/UWAC/uwac_impl.py:257
↓ 1 callersMethodcompute_loss
( self, batch: TorchMiniBatch, q_tpn: torch.Tensor, )
offline-rl-algorithms/E2O/d3rlpy_new/d3rlpy/ope/torch/fqe_impl.py:117
↓ 1 callersMethodcompute_loss
( self, obs_t: torch.Tensor, act_t: torch.Tensor )
offline-rl-algorithms/E2O/d3rlpy_new/d3rlpy/algos/torch/bc_impl.py:102
↓ 1 callersMethodcompute_loss
( self, batch: TorchMiniBatch, q_tpn: torch.Tensor, )
offline-rl-algorithms/E2O/d3rlpy_new/d3rlpy/algos/torch/dqn_impl.py:113
↓ 1 callersFunctioncompute_max_with_n_actions
( x: torch.Tensor, actions: torch.Tensor, q_func: EnsembleContinuousQFunction, lam: float, )
offline-rl-algorithms/E2O/d3rlpy_new/d3rlpy/models/torch/q_functions/__init__.py:66
↓ 1 callersMethodcompute_output_tensor
Returns the resulting tensor when all operations of this layer are applied to input_tensor. If input_tensor is 2D, this method returns a 3D t
modelbased-rl/BMPO/models/fc.py:83
↓ 1 callersMethodcompute_q_std
( self, x: torch.Tensor, action: Optional[torch.Tensor] = None, )
offline-rl-algorithms/E2O/d3rlpy_new/d3rlpy/models/torch/q_functions/ensemble_q_function.py:213
↓ 1 callersFunctioncompute_quantile_huber_loss
( y: torch.Tensor, target: torch.Tensor, taus: torch.Tensor )
offline-rl-algorithms/E2O/d3rlpy_new/d3rlpy/models/torch/q_functions/utility.py:35
↓ 1 callersMethodcompute_return
Computes sum of rewards. .. math:: R = \sum_{i=1} r_i Returns: flo
offline-rl-algorithms/E2O/d3rlpy_new/d3rlpy/dataset.py:99
↓ 1 callersMethodcompute_target
(self, batch: TorchMiniBatch)
offline-rl-algorithms/E2O/d3rlpy_new/d3rlpy/algos/torch/sac_impl.py:151
↓ 1 callersMethodcompute_target
(self, batch: TorchMiniBatch)
offline-rl-algorithms/E2O/d3rlpy_new/d3rlpy/algos/torch/sac_impl.py:338
↓ 1 callersMethodcompute_target
(self, batch: TorchMiniBatch)
offline-rl-algorithms/E2O/d3rlpy_new/d3rlpy/algos/torch/cql_impl.py:225
↓ 1 callersMethodcompute_target
(self, batch: TorchMiniBatch)
offline-rl-algorithms/E2O/d3rlpy_new/d3rlpy/algos/torch/ddpg_impl.py:190
↓ 1 callersMethodcompute_target
(self, batch: TorchMiniBatch)
offline-rl-algorithms/E2O/d3rlpy_new/d3rlpy/algos/torch/plas_impl.py:154
↓ 1 callersMethodcompute_target
(self, batch: TorchMiniBatch)
offline-rl-algorithms/E2O/d3rlpy_new/d3rlpy/algos/torch/e2o_impl.py:137
↓ 1 callersMethodcompute_target
( self, x: torch.Tensor, action: Optional[torch.Tensor] )
offline-rl-algorithms/E2O/d3rlpy_new/d3rlpy/models/torch/q_functions/base.py:34
↓ 1 callersMethodcompute_target
(self, batch: TorchMiniBatch)
offline-rl-algorithms/REDQ/redq_impl.py:309
↓ 1 callersMethodcompute_target
(self, batch: TorchMiniBatch)
offline-rl-algorithms/COMBO/t.py:148
↓ 1 callersMethodcompute_target
(self, batch: TorchMiniBatch)
offline-rl-algorithms/COMBO/t.py:309
↓ 1 callersMethodcompute_target_and_vars
(self, batch: TorchMiniBatch)
offline-rl-algorithms/UWAC/uwac_impl.py:329
↓ 1 callersMethodcompute_value_loss
(self, batch: TorchMiniBatch)
offline-rl-algorithms/E2O/d3rlpy_new/d3rlpy/algos/torch/iql_impl.py:162
↓ 1 callersMethodcompute_weighted_sac_actor_loss
(self, batch: TorchMiniBatch)
offline-rl-algorithms/UWAC/uwac_impl.py:197
↓ 1 callersFunctionconstruct_backward_model
(obs_dim=11, act_dim=3, rew_dim=1, hidden_dim=200, num_networks=7, num_elites=5, session=None)
modelbased-rl/BMPO/models/constructor.py:21
↓ 1 callersMethodconstruct_vars
Constructs the variables of this fully-connected layer. Returns: None
modelbased-rl/BMPO/models/fc.py:132
↓ 1 callersMethodconvert_D4RL
(self, dataset, reward_scale=1., reward_bias=0., standardize=False)
offline-rl-algorithms/ISPI/utils.py:40
↓ 1 callersFunctioncopy_source_tree
(source_dir, target_dir)
modelbased-rl/PlaNet/planet/scripts/sync.py:29
↓ 1 callersFunctioncreate_animation
(frames, size, fps=10, **kwargs)
modelbased-rl/PlaNet/planet/scripts/create_video.py:29
↓ 1 callersMethodcreate_continuous
Returns PyTorch's Q function module. Args: encoder: an encoder module that processes the observation and action t
offline-rl-algorithms/E2O/d3rlpy_new/d3rlpy/models/q_functions.py:45
↓ 1 callersFunctioncreate_deterministic_regressor
( observation_shape: Sequence[int], action_size: int, encoder_factory: EncoderFactory, )
offline-rl-algorithms/E2O/d3rlpy_new/d3rlpy/models/builders.py:152
↓ 1 callersMethodcreate_discrete
Returns PyTorch's Q function module. Args: encoder: an encoder module that processes the observation to obtain fe
offline-rl-algorithms/E2O/d3rlpy_new/d3rlpy/models/q_functions.py:29
↓ 1 callersFunctioncreate_probabilistic_ensemble_dynamics_model
( observation_shape: Sequence[int], action_size: int, encoder_factory: EncoderFactory, n_ensem
offline-rl-algorithms/E2O/d3rlpy_new/d3rlpy/models/builders.py:181
↓ 1 callersFunctioncreate_probablistic_regressor
( observation_shape: Sequence[int], action_size: int, encoder_factory: EncoderFactory, min_log
offline-rl-algorithms/E2O/d3rlpy_new/d3rlpy/models/builders.py:161
↓ 1 callersFunctioncreate_reader
(logdir)
modelbased-rl/PlaNet/planet/scripts/fetch_events.py:45
↓ 1 callersFunctioncreate_trial_name_creator
(trial_name_template=None)
modelbased-rl/MBPO/ED2-MBPO/examples/instrument.py:96
↓ 1 callersFunctioncreate_value_function
( observation_shape: Sequence[int], encoder_factory: EncoderFactory )
offline-rl-algorithms/E2O/d3rlpy_new/d3rlpy/models/builders.py:174
↓ 1 callersFunctioncurve_individual
(index, method, ax, runs, args)
modelbased-rl/Dreamer/Vanilla_Dreamer/plotting.py:173
↓ 1 callersFunctioncurve_std
(index, method, ax, runs, args)
modelbased-rl/Dreamer/Vanilla_Dreamer/plotting.py:184
↓ 1 callersMethoddecode_without_squash
( self, x: torch.Tensor, latent: torch.Tensor )
offline-rl-algorithms/E2O/d3rlpy_new/d3rlpy/models/torch/imitators.py:74
↓ 1 callersFunctiondefault
(config, params)
modelbased-rl/PlaNet/planet/scripts/configs.py:41
↓ 1 callersFunctiondefine_batch_env
(env_ctor, num_agents, isolate_envs)
modelbased-rl/PlaNet/planet/control/simulate.py:94
↓ 1 callersFunctiondefine_config
()
modelbased-rl/Dreamer/ED2-Dreamer/dreamer.py:54
↓ 1 callersFunctiondefine_config
()
modelbased-rl/Dreamer/Vanilla_Dreamer/dreamer.py:28
↓ 1 callersFunctiondetect_monitor_files
(training_dir)
multiagent-rl/easy-marl/envs/discrete_magym/wrappers/monitor.py:94
↓ 1 callersFunctiondetect_training_manifests
(training_dir, files=None)
multiagent-rl/easy-marl/envs/discrete_magym/wrappers/monitor.py:88
↓ 1 callersMethoddevice
(self)
offline-rl-algorithms/E2O/d3rlpy_new/d3rlpy/algos/torch/base.py:212
↓ 1 callersFunctiondict2obj
(dictObj)
multiagent-rl/easy-marl/utils/read_yaml.py:17
↓ 1 callersFunctiondict_to_cpu
(dictionary)
modelbased-rl/SampledMuZero/models.py:28
↓ 1 callersFunctiondisable_parallel
()
offline-rl-algorithms/E2O/d3rlpy_new/d3rlpy/context.py:19
↓ 1 callersFunctiondraw_border
(image, border_width=1, fill='black')
multiagent-rl/easy-marl/envs/discrete_magym/envs/utils/draw.py:107
↓ 1 callersFunctiondraw_cell_outline
(image, pos, cell_size=50, fill='black')
multiagent-rl/easy-marl/envs/discrete_magym/envs/utils/draw.py:90
↓ 1 callersMethoddynamics
(self, encoded_state, action)
modelbased-rl/SampledMuZero/models.py:131
↓ 1 callersMethoddynamics
(self, state, action)
modelbased-rl/MuZero/core/model.py:29
↓ 1 callersFunctionencode_gif
(frames, fps)
modelbased-rl/Dreamer/ED2-Dreamer/tools.py:82
↓ 1 callersFunctionencode_gif
(frames, fps)
modelbased-rl/Dreamer/Vanilla_Dreamer/tools.py:81
↓ 1 callersFunctionencode_gif
Encodes numpy images into gif string. Args: images: A 5-D `uint8` `np.array` (or a list of 4-D images) of shape `[batch_size, time, heigh
modelbased-rl/PlaNet/planet/tools/gif_summary.py:24
↓ 1 callersFunctionencoder
Extract deterministic features from an observation.
modelbased-rl/PlaNet/planet/networks/conv_ha.py:26
↓ 1 callersMethodencoder
(self)
offline-rl-algorithms/E2O/d3rlpy_new/d3rlpy/models/torch/q_functions/base.py:43
↓ 1 callersMethodend_episode
(self, agent_indices)
modelbased-rl/PlaNet/planet/control/mpc_agent.py:88
↓ 1 callersFunctionenv_ctor
()
modelbased-rl/PlaNet/planet/training/utility.py:281
↓ 1 callersFunctioneval_policy
(policy, env_name, seed, mean, std, seed_offset=100, eval_episodes=10)
offline-rl-algorithms/ISPI/main.py:19
↓ 1 callersFunctionevaluate_policy
(env, agent, max_episode_steps, deterministic=True)
offline-rl-algorithms/E2O/PEX-main/pex/utils/util.py:278
↓ 1 callersMethodexpanded
(self)
modelbased-rl/SampledMuZero/self_play.py:439
↓ 1 callersMethodexpanded
(self)
modelbased-rl/MuZero/core/mcts.py:36
↓ 1 callersFunctionexpectile_loss
(diff, expectile)
offline-rl-algorithms/E2O/PEX-main/pex/algorithms/iql.py:14
↓ 1 callersMethodexperience
(self, agent_indices, *experience)
modelbased-rl/PlaNet/planet/control/mpc_agent.py:85
↓ 1 callersFunctionexploration
(q_list, epsilon, args)
multiagent-rl/easy-marl/main_dqn.py:22
↓ 1 callersFunctionexport_scalar
(basename, steps, times, values)
modelbased-rl/PlaNet/planet/scripts/fetch_events.py:60
↓ 1 callersFunctionextract_values
(reader, tag)
modelbased-rl/PlaNet/planet/scripts/fetch_events.py:52
↓ 1 callersFunctionfigure
(runs, args)
modelbased-rl/Dreamer/Vanilla_Dreamer/plotting.py:110
↓ 1 callersMethodfinalize
Finalizes the network. Arguments: optimizer: (tf.train.Optimizer) An optimizer class from those available at tf.train.Optimizer.
modelbased-rl/MBPO/ED2-MBPO/mbpo/models/bnn.py:172
↓ 1 callersFunctionfind_keys
(args)
modelbased-rl/Dreamer/Vanilla_Dreamer/plotting.py:31
↓ 1 callersMethodfit
Runs two ops, one for assigning the mean of the data to the internal mean, and another for assigning the standard deviation of the data to the
modelbased-rl/MBPO/ED2-MBPO/mbpo/models/utils.py:38
↓ 1 callersMethodfit
Runs two ops, one for assigning the mean of the data to the internal mean, and another for assigning the standard deviation of the data to the
modelbased-rl/BMPO/models/utils.py:38
↓ 1 callersMethodfit
(self, transitions: List[Transition])
offline-rl-algorithms/E2O/d3rlpy_new/d3rlpy/preprocessing/reward_scalers.py:252
↓ 1 callersMethodfit
(self, transitions: List[Transition])
offline-rl-algorithms/E2O/d3rlpy_new/d3rlpy/preprocessing/reward_scalers.py:347
↓ 1 callersMethodfit
(self, transitions: List[Transition])
offline-rl-algorithms/E2O/d3rlpy_new/d3rlpy/preprocessing/reward_scalers.py:442
↓ 1 callersMethodfit
(self, transitions: List[Transition])
offline-rl-algorithms/E2O/d3rlpy_new/d3rlpy/preprocessing/action_scalers.py:161
↓ 1 callersMethodfit
(self, transitions: List[Transition])
offline-rl-algorithms/E2O/d3rlpy_new/d3rlpy/preprocessing/scalers.py:188
↓ 1 callersMethodfit
(self, transitions: List[Transition])
offline-rl-algorithms/E2O/d3rlpy_new/d3rlpy/preprocessing/scalers.py:325
↓ 1 callersMethodfit_online
Start training loop of online deep reinforcement learning. Args: env: gym-like environment. buffer : replay buffer.
offline-rl-algorithms/E2O/d3rlpy_new/d3rlpy/algos/base.py:203
← previousnext →1,001–1,100 of 3,230, ranked by callers