MCPcopy Create free account

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

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

↓ 2 callersMethodbuild
(self)
offline-rl-algorithms/E2O/d3rlpy_new/d3rlpy/ope/torch/fqe_impl.py:79
↓ 2 callersMethodbuild
(self)
offline-rl-algorithms/REDQ/redq_impl.py:92
↓ 2 callersMethodbuild
(self)
offline-rl-algorithms/COMBO/t.py:92
↓ 2 callersMethodcall
(self, name, *args, **kwargs)
modelbased-rl/Dreamer/ED2-Dreamer/wrappers.py:419
↓ 2 callersMethodcall
Asynchronously call a method of the external environment. Args: name: Name of the method to call. *args: Positional arguments to forw
modelbased-rl/PlaNet/planet/control/wrappers.py:625
↓ 2 callersMethodclose
(self)
modelbased-rl/BMPO/utils/logging.py:140
↓ 2 callersFunctioncompute_huber_loss
( y: torch.Tensor, target: torch.Tensor, beta: float = 1.0 )
offline-rl-algorithms/E2O/d3rlpy_new/d3rlpy/models/torch/q_functions/utility.py:27
↓ 2 callersMethodcompute_log_probs_with_logits
( self, x: torch.Tensor )
offline-rl-algorithms/E2O/d3rlpy_new/d3rlpy/models/torch/imitators.py:150
↓ 2 callersFunctioncompute_max_with_n_actions_and_indices
Returns weighted target value from sampled actions. This calculation is proposed in BCQ paper for the first time. `x` should be shaped with `(
offline-rl-algorithms/E2O/d3rlpy_new/d3rlpy/models/torch/q_functions/__init__.py:8
↓ 2 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/MBPO/ED2-MBPO/mbpo/models/fc.py:83
↓ 2 callersMethodcompute_stats
( self, x: torch.Tensor, action: torch.Tensor )
offline-rl-algorithms/E2O/d3rlpy_new/d3rlpy/models/torch/dynamics.py:83
↓ 2 callersMethodcompute_target
( self, batch: TorchMiniBatch, next_actions: torch.Tensor )
offline-rl-algorithms/E2O/d3rlpy_new/d3rlpy/ope/torch/fqe_impl.py:132
↓ 2 callersMethodcompute_target
(self, batch: TorchMiniBatch)
offline-rl-algorithms/E2O/d3rlpy_new/d3rlpy/algos/torch/dqn_impl.py:128
↓ 2 callersMethodcompute_target_redq
( self, x: torch.Tensor, action: Optional[torch.Tensor] = None, reduction: str
offline-rl-algorithms/E2O/d3rlpy_new/d3rlpy/models/torch/q_functions/ensemble_q_function.py:257
↓ 2 callersMethodcompute_target_value
(self, game_history, index)
modelbased-rl/SampledMuZero/replay_buffer.py:222
↓ 2 callersMethodcompute_target_weighted
( self, x: torch.Tensor, action: torch.Tensor, reduction: str = "min",
offline-rl-algorithms/E2O/d3rlpy_new/d3rlpy/models/torch/q_functions/ensemble_q_function.py:324
↓ 2 callersFunctionconfirm_yes_no
(prompt)
modelbased-rl/MBPO/ED2-MBPO/examples/instrument.py:153
↓ 2 callersFunctionconstruct_forward_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:8
↓ 2 callersFunctionconstruct_model
(obs_dim=11, act_dim=3, rew_dim=1, hidden_dim=200, num_networks=7, num_elites=5, session=None, action_group =
modelbased-rl/MBPO/ED2-MBPO/mbpo/models/constructor.py:7
↓ 2 callersMethodconstruct_vars
Constructs the variables of this fully-connected layer. Returns: None
modelbased-rl/MBPO/ED2-MBPO/mbpo/models/fc.py:132
↓ 2 callersMethodcontains
contains observation
multiagent-rl/easy-marl/envs/discrete_magym/envs/utils/observation_space.py:16
↓ 2 callersMethodcreate
(self, observation_shape: Sequence[int])
offline-rl-algorithms/E2O/d3rlpy_new/d3rlpy/models/encoders.py:209
↓ 2 callersFunctioncreate_action_scaler
Returns registered action scaler object. Args: name: regsitered scaler type name. kwargs: scaler arguments. Returns:
offline-rl-algorithms/E2O/d3rlpy_new/d3rlpy/preprocessing/action_scalers.py:245
↓ 2 callersFunctioncreate_continuous_q_function
( observation_shape: Sequence[int], action_size: int, encoder_factory: EncoderFactory, q_func_
offline-rl-algorithms/E2O/d3rlpy_new/d3rlpy/models/builders.py:47
↓ 2 callersFunctioncreate_deterministic_residual_policy
( observation_shape: Sequence[int], action_size: int, scale: float, encoder_factory: EncoderFa
offline-rl-algorithms/E2O/d3rlpy_new/d3rlpy/models/builders.py:81
↓ 2 callersFunctioncreate_discrete_imitator
( observation_shape: Sequence[int], action_size: int, beta: float, encoder_factory: EncoderFac
offline-rl-algorithms/E2O/d3rlpy_new/d3rlpy/models/builders.py:142
↓ 2 callersFunctioncreate_encoder_factory
Returns registered encoder factory object. Args: name: regsitered encoder factory type name. kwargs: encoder arguments. Retu
offline-rl-algorithms/E2O/d3rlpy_new/d3rlpy/models/encoders.py:422
↓ 2 callersFunctioncreate_q_func_factory
Returns registered Q function factory object. Args: name: registered Q function factory type name. kwargs: Q function arguments.
offline-rl-algorithms/E2O/d3rlpy_new/d3rlpy/models/q_functions.py:339
↓ 2 callersFunctioncreate_reward_scaler
(name: str, **kwargs: Any)
offline-rl-algorithms/E2O/d3rlpy_new/d3rlpy/preprocessing/reward_scalers.py:505
↓ 2 callersFunctioncreate_scaler
Returns registered scaler object. Args: name: regsitered scaler type name. kwargs: scaler arguments. Returns: scaler
offline-rl-algorithms/E2O/d3rlpy_new/d3rlpy/preprocessing/scalers.py:396
↓ 2 callersMethodcreate_with_action
( self, observation_shape: Sequence[int], action_size: int, discrete_action: b
offline-rl-algorithms/E2O/d3rlpy_new/d3rlpy/models/encoders.py:220
↓ 2 callersMethoddisable
(self)
multiagent-rl/easy-marl/envs/continuous_mpe/multiagent/rendering.py:173
↓ 2 callersMethoddist
(self, x: torch.Tensor)
offline-rl-algorithms/E2O/d3rlpy_new/d3rlpy/models/torch/imitators.py:207
↓ 2 callersMethoddist_from_state
Extract the latent distribution from a prior or posterior state.
modelbased-rl/PlaNet/planet/models/rssm.py:74
↓ 2 callersMethoddist_from_state
Extract the latent distribution from a prior or posterior state.
modelbased-rl/PlaNet/planet/models/drnn.py:79
↓ 2 callersMethoddraw_circle
(self, radius=10, res=30, filled=True, **attrs)
multiagent-rl/easy-marl/envs/continuous_mpe/multiagent/rendering.py:115
↓ 2 callersMethoddraw_polyline
(self, v, **attrs)
multiagent-rl/easy-marl/envs/continuous_mpe/multiagent/rendering.py:127
↓ 2 callersMethodenable
(self)
multiagent-rl/easy-marl/envs/continuous_mpe/multiagent/rendering.py:171
↓ 2 callersFunctioneval_policy
(env, env_name, alg, max_episode_steps, n_eval_episodes, seed)
offline-rl-algorithms/E2O/PEX-main/pex/utils/util.py:263
↓ 2 callersMethodfit_with_env
Gets scaling parameters from environment. Args: env: gym environment.
offline-rl-algorithms/E2O/d3rlpy_new/d3rlpy/preprocessing/scalers.py:26
↓ 2 callersFunctionflatten_
Combine all leaves of a nested structure into a tuple. The nested structure can consist of any combination of tuples, lists, and dicts. Dictionar
modelbased-rl/PlaNet/planet/tools/nested.py:95
↓ 2 callersMethodforward
(self, x)
offline-rl-algorithms/E2O/PEX-main/pex/networks/policy.py:63
↓ 2 callersMethodforward
( self, x: torch.Tensor, deterministic: bool = False, with_log_prob: bool = Fa
offline-rl-algorithms/E2O/d3rlpy_new/d3rlpy/models/torch/policies.py:183
↓ 2 callersMethodforward
( self, x: torch.Tensor, deterministic: bool = False, with_log_prob: bool = Fa
offline-rl-algorithms/E2O/d3rlpy_new/d3rlpy/models/torch/policies.py:277
↓ 2 callersMethodforward
(self, x: torch.Tensor, action: torch.Tensor)
offline-rl-algorithms/E2O/d3rlpy_new/d3rlpy/models/torch/q_functions/mean_q_function.py:71
↓ 2 callersMethodfoward_agent_i
(self, observation, agent_id)
multiagent-rl/easy-marl/algorithms/PPO_based/IPPO.py:44
↓ 2 callersMethodgenerate_action
(self, observation_list)
multiagent-rl/easy-marl/algorithms/DDPG_based/IDDPG.py:99
↓ 2 callersMethodgenerate_action_list
(self, observation_list)
multiagent-rl/easy-marl/algorithms/PPO_based/IPPO.py:74
↓ 2 callersMethodgenerate_action_prob_V_agent_i
(self, observation, agent_id)
multiagent-rl/easy-marl/algorithms/PPO_based/IPPO.py:83
↓ 2 callersMethodgenerate_action_prob_V_agent_i
(self, observation, agent_id, state)
multiagent-rl/easy-marl/algorithms/PPO_based/MAPPO.py:121
↓ 2 callersMethodgenerate_q_list
(self, observation_list)
multiagent-rl/easy-marl/algorithms/DQN_based/VDN.py:78
↓ 2 callersFunctionget_action_size_from_env
(env: gym.Env)
offline-rl-algorithms/E2O/d3rlpy_new/d3rlpy/online/utility.py:7
↓ 2 callersMethodget_action_type
(self)
offline-rl-algorithms/E2O/d3rlpy_new/d3rlpy/algos/bc.py:160
↓ 2 callersMethodget_activation
Returns the current activation function for this layer. Arguments: as_func: (bool) Determines whether the returned value is the s
modelbased-rl/MBPO/ED2-MBPO/mbpo/models/fc.py:205
↓ 2 callersMethodget_activation
Returns the current activation function for this layer. Arguments: as_func: (bool) Determines whether the returned value is the s
modelbased-rl/BMPO/models/fc.py:205
↓ 2 callersMethodget_agent_obs
(self)
multiagent-rl/easy-marl/envs/discrete_magym/envs/predator_prey/predator_prey.py:114
↓ 2 callersMethodget_agent_obs
(self)
multiagent-rl/easy-marl/envs/discrete_magym/envs/switch/switch_one_corridor.py:90
↓ 2 callersMethodget_agent_obs
(self)
multiagent-rl/easy-marl/envs/discrete_magym/envs/checkers/checkers.py:100
↓ 2 callersMethodget_agent_obs
Computes the observations for the agents. Each agent receives information about cars in it's vision range (a surrounding 3 × 3 neighb
multiagent-rl/easy-marl/envs/discrete_magym/envs/traffic_junction/traffic_junction.py:210
↓ 2 callersMethodget_agent_obs
(self)
multiagent-rl/easy-marl/envs/discrete_magym/envs/pong_duel/pong_duel.py:84
↓ 2 callersFunctionget_batch_from_buffer
(memory, batch_size)
offline-rl-algorithms/E2O/PEX-main/pex/utils/util.py:137
↓ 2 callersFunctionget_cartpole
Returns cartpole dataset and environment. The dataset is automatically downloaded to ``d3rlpy_data/cartpole.h5`` if it does not exist. A
offline-rl-algorithms/E2O/d3rlpy_new/d3rlpy/datasets.py:23
↓ 2 callersFunctionget_d4rl
Returns d4rl dataset and envrironment. The dataset is provided through d4rl. .. code-block:: python from d3rlpy.datasets import get_d4
offline-rl-algorithms/E2O/d3rlpy_new/d3rlpy/datasets.py:213
↓ 2 callersMethodget_decays
Returns the list of losses corresponding to the weight decay imposed on each weight of the network. Returns: the list of weight decay
modelbased-rl/MBPO/ED2-MBPO/mbpo/models/fc.py:106
↓ 2 callersMethodget_diagnostics
Return diagnostic information as ordered dictionary. Records mean and standard deviation of Q-function and state value function, and
modelbased-rl/MBPO/ED2-MBPO/mbpo/algorithms/mbpo.py:688
↓ 2 callersMethodget_diagnostics
Return diagnostic information as ordered dictionary. Records mean and standard deviation of Q-function and state value function, and
modelbased-rl/BMPO/bmpo.py:765
↓ 2 callersMethodget_dist
(self, state)
modelbased-rl/Dreamer/ED2-Dreamer/models.py:137
↓ 2 callersFunctionget_env_and_dataset
(env_name, max_episode_steps)
offline-rl-algorithms/E2O/PEX-main/pex/utils/util.py:244
↓ 2 callersFunctionget_experiments_info
(experiments)
modelbased-rl/MBPO/ED2-MBPO/examples/instrument.py:135
↓ 2 callersMethodget_logstd_parameter
(self)
offline-rl-algorithms/E2O/d3rlpy_new/d3rlpy/models/torch/policies.py:255
↓ 2 callersMethodget_model_vars
(self, idx, sess)
modelbased-rl/MBPO/ED2-MBPO/mbpo/models/fc.py:54
↓ 2 callersMethodget_next
(self)
offline-rl-algorithms/E2O/d3rlpy_new/d3rlpy/iterators/base.py:87
↓ 2 callersMethodget_observation
(self)
modelbased-rl/SampledMuZero/games/simple_grid.py:224
↓ 2 callersFunctionget_pendulum
Returns pendulum dataset and environment. The dataset is automatically downloaded to ``d3rlpy_data/pendulum.h5`` if it does not exist. A
offline-rl-algorithms/E2O/d3rlpy_new/d3rlpy/datasets.py:63
↓ 2 callersFunctionget_plt
()
offline-rl-algorithms/E2O/d3rlpy_new/d3rlpy/cli.py:32
↓ 2 callersFunctionget_state_dict
(impl: Any)
offline-rl-algorithms/E2O/d3rlpy_new/d3rlpy/torch_utility.py:107
↓ 2 callersFunctionget_variant_spec_base
(universe, domain, task, policy, algorithm)
modelbased-rl/MBPO/ED2-MBPO/examples/development/variants.py:236
↓ 2 callersMethodget_vars
Returns the variables of this layer.
modelbased-rl/BMPO/models/fc.py:158
↓ 2 callersMethodget_weights
(self)
modelbased-rl/SampledMuZero/models.py:53
↓ 2 callersMethodgood_agents
(self, world)
multiagent-rl/easy-marl/envs/continuous_mpe/multiagent/scenarios/simple_tag.py:76
↓ 2 callersMethodimg_step
(self, prev_state, prev_action)
modelbased-rl/Dreamer/ED2-Dreamer/models.py:70
↓ 2 callersMethodimg_step
(self, prev_state, prev_action)
modelbased-rl/Dreamer/Vanilla_Dreamer/models.py:70
↓ 2 callersFunctionimpl
(function, *structures)
modelbased-rl/PlaNet/planet/tools/nested.py:69
↓ 2 callersFunctioninfer_headers
(directory)
modelbased-rl/PlaNet/planet/scripts/sync.py:45
↓ 2 callersMethodinitial
(self, batch_size)
modelbased-rl/Dreamer/ED2-Dreamer/models.py:101
↓ 2 callersMethodinverse_reward_transform
(self, reward_logits)
modelbased-rl/MuZero/core/config.py:120
↓ 2 callersMethodinverse_scalar_transform
Reference : Appendix F => Network Architecture & Appendix A : Proposition A.2 in https://arxiv.org/pdf/1805.11593.pdf (Page-11)
modelbased-rl/MuZero/core/config.py:126
↓ 2 callersMethodis_collision
(self, agent1, agent2)
multiagent-rl/easy-marl/envs/continuous_mpe/multiagent/scenarios/simple_spread.py:65
↓ 2 callersMethodis_valid
(self, pos)
multiagent-rl/easy-marl/envs/discrete_magym/envs/combat/combat.py:323
↓ 2 callersMethodis_valid
(self, pos)
multiagent-rl/easy-marl/envs/discrete_magym/envs/checkers/checkers.py:148
↓ 2 callersMethodis_valid
(self, pos)
multiagent-rl/easy-marl/envs/discrete_magym/envs/traffic_junction/traffic_junction.py:165
↓ 2 callersMethoditerate
Run the schedule for a specified number of steps and yield scores. Call the operation of the current phase until the global step reaches the
modelbased-rl/PlaNet/planet/training/trainer.py:162
↓ 2 callersFunctionlaunch_example_cluster
Launches the example on autoscaled ray cluster through ray exec_cmd. This handles basic validation and sanity checks for the experiment, and
modelbased-rl/MBPO/ED2-MBPO/examples/instrument.py:288
↓ 2 callersMethodlegal_actions
Should return the legal actions at each turn, if it is not available, it can return the whole action space. At each turn, the game ha
modelbased-rl/SampledMuZero/games/atari.py:161
↓ 2 callersMethodload
(self, filename)
modelbased-rl/Dreamer/Vanilla_Dreamer/tools.py:30
↓ 2 callersMethodlock
(self)
modelbased-rl/PlaNet/planet/tools/attr_dict.py:95
↓ 2 callersMethodloss_function
( value, reward, policy_logits, target_value, target_reward, t
modelbased-rl/SampledMuZero/trainer.py:284
↓ 2 callersFunctionmain
()
modelbased-rl/MBPO/ED2-MBPO/mbpo/scripts/console_scripts.py:11
↓ 2 callersFunctionmake_env
(config, writer, prefix, datadir, store)
modelbased-rl/Dreamer/ED2-Dreamer/dreamer.py:416
↓ 2 callersFunctionmake_env
(config, writer, prefix, datadir, store)
modelbased-rl/Dreamer/Vanilla_Dreamer/dreamer.py:376
← previousnext →401–500 of 3,230, ranked by callers