MCPcopy Create free account

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

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

↓ 2 callersMethod__draw_base_img
(self)
multiagent-rl/easy-marl/envs/discrete_magym/envs/traffic_junction/traffic_junction.py:258
↓ 2 callersMethod__getattr__
(self, name)
modelbased-rl/Dreamer/ED2-Dreamer/wrappers.py:413
↓ 2 callersMethod__getattr__
(self, name)
modelbased-rl/Dreamer/Vanilla_Dreamer/wrappers.py:398
↓ 2 callersMethod__getattr__
Request an attribute from the environment. Note that this involves communication with the external process, so it can be slow. Args:
modelbased-rl/PlaNet/planet/control/wrappers.py:610
↓ 2 callersMethod__init__
(self, seed=None)
modelbased-rl/SampledMuZero/games/lunarlander.py:131
↓ 2 callersMethod__init__
(self, args)
multiagent-rl/easy-marl/algorithms/DDPG_based/IDDPG.py:91
↓ 2 callersMethod__init__
(self, args)
multiagent-rl/easy-marl/algorithms/DDPG_based/MADDPG.py:93
↓ 2 callersMethod__init__
(self, args)
multiagent-rl/easy-marl/algorithms/PPO_based/MAPPO.py:106
↓ 2 callersMethod__init__
(self, cache_size=1)
offline-rl-algorithms/E2O/PEX-main/pex/networks/policy.py:142
↓ 2 callersMethod__init__
( self, *, learning_rate: float = 1e-3, optim_factory: OptimizerFactory = Adam
offline-rl-algorithms/E2O/d3rlpy_new/d3rlpy/algos/bc.py:115
↓ 2 callersMethod__init__
( self, observation_shape: Sequence[int], action_size: int, learning_rate: flo
offline-rl-algorithms/E2O/d3rlpy_new/d3rlpy/algos/torch/bc_impl.py:123
↓ 2 callersMethod__init__
( self, momentum: float = 0, dampening: float = 0, weight_decay: float = 0,
offline-rl-algorithms/E2O/d3rlpy_new/d3rlpy/models/optimizers.py:89
↓ 2 callersMethod__init__
( self, state_dim, action_dim, max_action, di
offline-rl-algorithms/ISPI/ISPI_S.py:64
↓ 2 callersMethod__init__
( self, state_dim, action_dim, max_action, de
offline-rl-algorithms/ISPI/ISPI_C.py:47
↓ 2 callersMethod__init_ball_pos
(self)
multiagent-rl/easy-marl/envs/discrete_magym/envs/pong_duel/pong_duel.py:103
↓ 2 callersMethod__init_full_obs
(self)
multiagent-rl/easy-marl/envs/discrete_magym/envs/switch/switch_one_corridor.py:83
↓ 2 callersMethod__update_agent_view
(self, agent_i)
multiagent-rl/easy-marl/envs/discrete_magym/envs/predator_prey/predator_prey.py:220
↓ 2 callersMethod__update_agent_view
(self, agent_i)
multiagent-rl/easy-marl/envs/discrete_magym/envs/combat/combat.py:180
↓ 2 callersMethod__update_agent_view
(self, agent_i)
multiagent-rl/easy-marl/envs/discrete_magym/envs/switch/switch_one_corridor.py:144
↓ 2 callersMethod__update_agent_view
(self, agent_i)
multiagent-rl/easy-marl/envs/discrete_magym/envs/checkers/checkers.py:174
↓ 2 callersMethod__update_opp_view
(self, opp_i)
multiagent-rl/easy-marl/envs/discrete_magym/envs/combat/combat.py:184
↓ 2 callersMethod__update_prey_view
(self, prey_i)
multiagent-rl/easy-marl/envs/discrete_magym/envs/predator_prey/predator_prey.py:223
↓ 2 callersFunction_active_collection
(collects, defaults, config, params)
modelbased-rl/PlaNet/planet/scripts/configs.py:246
↓ 2 callersMethod_agent_generator
Yields agent_id and agent for all agents in environment.
multiagent-rl/easy-marl/envs/discrete_magym/envs/lumberjacks/lumberjacks.py:258
↓ 2 callersMethod_claim
Ensure that no other worker will pick up this run or raise an exception. Note that usually the last worker who claims a run wins, since its name
modelbased-rl/PlaNet/planet/training/running.py:252
↓ 2 callersMethod_compile_losses
Helper method for compiling the loss function. The loss function is obtained from the log likelihood, assuming that the output distri
modelbased-rl/MBPO/ED2-MBPO/mbpo/models/bnn.py:667
↓ 2 callersMethod_compile_losses
Helper method for compiling the loss function. The loss function is obtained from the log likelihood, assuming that the output distri
modelbased-rl/BMPO/models/bnn.py:570
↓ 2 callersMethod_compile_outputs
Compiles the output of the network at the given inputs. If inputs is 2D, returns a 3D tensor where output[i] is the output of the ith network
modelbased-rl/MBPO/ED2-MBPO/mbpo/models/bnn.py:631
↓ 2 callersMethod_compile_outputs
Compiles the output of the network at the given inputs. If inputs is 2D, returns a 3D tensor where output[i] is the output of the ith network
modelbased-rl/BMPO/models/bnn.py:540
↓ 2 callersMethod_compute_conservative_loss
( # CQL总loss的前半部分 = α * (logsumexp - data_values) self, obs_t: torch.Tensor, act_t: torch.Tensor, obs
offline-rl-algorithms/E2O/d3rlpy_new/d3rlpy/algos/torch/cql_impl.py:196
↓ 2 callersMethod_compute_random_is_values
(self, obs: torch.Tensor)
offline-rl-algorithms/E2O/d3rlpy_new/d3rlpy/algos/torch/cql_impl.py:173
↓ 2 callersMethod_compute_target
( 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:113
↓ 2 callersMethod_compute_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:141
↓ 2 callersMethod_convert
(self, value)
modelbased-rl/Dreamer/ED2-Dreamer/wrappers.py:201
↓ 2 callersMethod_convert
(self, value)
modelbased-rl/Dreamer/Vanilla_Dreamer/wrappers.py:186
↓ 2 callersMethod_create_particle
(self, mass, x, y, ttl)
modelbased-rl/SampledMuZero/games/lunarlander.py:443
↓ 2 callersMethod_do_training_repeats
Repeat training _n_train_repeat times every _train_every_n_steps
modelbased-rl/BMPO/bmpo.py:718
↓ 2 callersFunction_exec_to_create_env
(code: str)
offline-rl-algorithms/E2O/d3rlpy_new/d3rlpy/cli.py:249
↓ 2 callersMethod_fc_encode
(self, x: torch.Tensor)
offline-rl-algorithms/E2O/d3rlpy_new/d3rlpy/models/torch/encoders.py:325
↓ 2 callersMethod_format_value
(self, value)
modelbased-rl/PlaNet/planet/tools/attr_dict.py:136
↓ 2 callersFunction_gaussian_likelihood
( x: torch.Tensor, mu: torch.Tensor, logstd: torch.Tensor )
offline-rl-algorithms/E2O/d3rlpy_new/d3rlpy/models/torch/dynamics.py:38
↓ 2 callersMethod_get_feed_dict
Construct TensorFlow feed_dict from sample batch.
modelbased-rl/MBPO/ED2-MBPO/mbpo/algorithms/mbpo.py:668
↓ 2 callersMethod_get_feed_dict
Construct TensorFlow feed_dict from sample batch.
modelbased-rl/BMPO/bmpo.py:745
↓ 2 callersMethod_get_last_conv_shape
(self)
offline-rl-algorithms/E2O/d3rlpy_new/d3rlpy/models/torch/encoders.py:134
↓ 2 callersMethod_get_linear_input_size
(self)
offline-rl-algorithms/E2O/d3rlpy_new/d3rlpy/models/torch/encoders.py:129
↓ 2 callersMethod_get_obs
(self)
modelbased-rl/MBPO/ED2-MBPO/mbpo/env/ant.py:36
↓ 2 callersMethod_get_obs
(self)
modelbased-rl/MBPO/ED2-MBPO/mbpo/env/humanoid.py:21
↓ 2 callersMethod_get_obs
(self)
modelbased-rl/Dreamer/ED2-Dreamer/wrappers.py:142
↓ 2 callersMethod_get_obs
(self)
modelbased-rl/Dreamer/Vanilla_Dreamer/wrappers.py:140
↓ 2 callersMethod_get_obs
(self)
modelbased-rl/BMPO/env/ant.py:27
↓ 2 callersMethod_get_obs
(self)
modelbased-rl/BMPO/env/walker2d.py:40
↓ 2 callersMethod_get_obs
(self)
modelbased-rl/BMPO/env/hopper.py:24
↓ 2 callersMethod_get_obs
(self)
modelbased-rl/BMPO/env/walker2dNT.py:39
↓ 2 callersMethod_get_obs
(self)
modelbased-rl/BMPO/env/hopperNT.py:26
↓ 2 callersMethod_get_obs
(self)
modelbased-rl/BMPO/env/pendulum.py:54
↓ 2 callersMethod_get_obs
(self)
multiagent-rl/easy-marl/envs/discrete_magym/ma_gym_env.py:53
↓ 2 callersMethod_get_obs
(self, agent)
multiagent-rl/easy-marl/envs/continuous_mpe/multiagent/environment.py:130
↓ 2 callersMethod_get_obs
(self)
offline-rl-algorithms/E2O/d3rlpy_new/d3rlpy/envs/wrappers.py:230
↓ 2 callersMethod_get_observation
(self, agent)
multiagent-rl/easy-marl/envs/discrete_meeting.py:62
↓ 2 callersMethod_get_observation
(self, agent)
multiagent-rl/easy-marl/envs/continuous_meeting.py:57
↓ 2 callersMethod_get_observation_list
(self)
multiagent-rl/easy-marl/envs/discrete_meeting.py:59
↓ 2 callersMethod_get_observation_list
(self)
multiagent-rl/easy-marl/envs/continuous_meeting.py:54
↓ 2 callersMethod_get_state
concat all agent's observation to construct state info
multiagent-rl/easy-marl/envs/discrete_meeting.py:71
↓ 2 callersMethod_get_state
concat all agent's observation to construct state info
multiagent-rl/easy-marl/envs/continuous_meeting.py:66
↓ 2 callersMethod_get_state
(self)
multiagent-rl/easy-marl/envs/discrete_magym/ma_gym_env.py:56
↓ 2 callersMethod_get_state
(self)
multiagent-rl/easy-marl/envs/continuous_mpe/mpe_env.py:66
↓ 2 callersMethod_get_tf_checkpoint
(self)
modelbased-rl/MBPO/ED2-MBPO/examples/development/main.py:103
↓ 2 callersMethod_initialize_variables
Initialize or restore variables from a checkpoint if available. Args: sess: Session to initialize variables in. savers: List of saver
modelbased-rl/PlaNet/planet/training/trainer.py:302
↓ 2 callersMethod_is_cell_vacant
(self, pos)
multiagent-rl/easy-marl/envs/discrete_magym/envs/combat/combat.py:326
↓ 2 callersFunction_make_taus
( h: torch.Tensor, n_quantiles: int, training: bool )
offline-rl-algorithms/E2O/d3rlpy_new/d3rlpy/models/torch/q_functions/iqn_q_function.py:16
↓ 2 callersFunction_merge_dims
Flatten consecutive axes of a tensor trying to preserve static shapes.
modelbased-rl/PlaNet/planet/tools/overshooting.py:145
↓ 2 callersMethod_one_hot
(self, indices)
modelbased-rl/Dreamer/ED2-Dreamer/tools.py:313
↓ 2 callersMethod_one_hot
(self, indices)
modelbased-rl/Dreamer/Vanilla_Dreamer/tools.py:286
↓ 2 callersMethod_parse_shape
Get a tensor shape from a OpenAI Gym space. Args: space: Gym space. Raises: NotImplementedError: For spaces other than Box and D
modelbased-rl/PlaNet/planet/control/in_graph_batch_env.py:150
↓ 2 callersFunction_permuted
(sequence, amount)
modelbased-rl/PlaNet/planet/tools/numpy_episodes.py:177
↓ 2 callersMethod_phi
(x, min, max, set_size: int)
modelbased-rl/MuZero/core/config.py:150
↓ 2 callersMethod_pickle_path
(self, checkpoint_dir)
modelbased-rl/MBPO/ED2-MBPO/examples/development/main.py:94
↓ 2 callersMethod_predict_best_action
(self, x: torch.Tensor)
offline-rl-algorithms/E2O/d3rlpy_new/d3rlpy/algos/torch/dqn_impl.py:139
↓ 2 callersMethod_predict_best_action
(self, x: torch.Tensor)
offline-rl-algorithms/E2O/d3rlpy_new/d3rlpy/algos/torch/base.py:66
↓ 2 callersMethod_process_observ
(self, observ)
modelbased-rl/PlaNet/planet/control/wrappers.py:491
↓ 2 callersMethod_process_reset
(self, observ)
modelbased-rl/PlaNet/planet/control/wrappers.py:482
↓ 2 callersMethod_process_step
(self, action, observ, reward, done, info)
modelbased-rl/PlaNet/planet/control/wrappers.py:469
↓ 2 callersMethod_render_image
(self)
modelbased-rl/PlaNet/planet/control/wrappers.py:166
↓ 2 callersMethod_repeat_observation
(self, x: torch.Tensor)
offline-rl-algorithms/E2O/d3rlpy_new/d3rlpy/algos/torch/bcq_impl.py:163
↓ 2 callersMethod_replay_pool_pickle_path
(self, checkpoint_dir)
modelbased-rl/MBPO/ED2-MBPO/examples/development/main.py:97
↓ 2 callersMethod_reset_render
(self)
multiagent-rl/easy-marl/envs/continuous_mpe/multiagent/environment.py:200
↓ 2 callersMethod_sample_repeated_action
( self, repeated_x: torch.Tensor, target: bool = False )
offline-rl-algorithms/E2O/d3rlpy_new/d3rlpy/algos/torch/bcq_impl.py:169
↓ 2 callersMethod_select_keys
(self, obs)
modelbased-rl/PlaNet/planet/control/wrappers.py:98
↓ 2 callersMethod_store_done
Mark run as finished by writing a DONE file.
modelbased-rl/PlaNet/planet/training/running.py:270
↓ 2 callersMethod_tf_checkpoint_prefix
(self, checkpoint_dir)
modelbased-rl/MBPO/ED2-MBPO/examples/development/main.py:100
↓ 2 callersMethod_train_target_network_hard
(self)
multiagent-rl/easy-marl/algorithms/DDPG_based/IDDPG.py:188
↓ 2 callersMethod_train_target_network_hard
(self)
multiagent-rl/easy-marl/algorithms/DDPG_based/MADDPG.py:190
↓ 2 callersMethod_train_target_network_hard
(self)
multiagent-rl/easy-marl/algorithms/DQN_based/your_new_algorithm.py:86
↓ 2 callersMethod_update_target
(self, tau=None)
modelbased-rl/MBPO/ED2-MBPO/mbpo/algorithms/mbpo.py:643
↓ 2 callersMethod_update_target
(self)
modelbased-rl/BMPO/bmpo.py:693
↓ 2 callersMethod_view_generator
Yields position, number of agent and tree strength for all cells defined by `mask`. Args: mask: tuple of slices in extended coord
multiagent-rl/easy-marl/envs/discrete_magym/envs/lumberjacks/lumberjacks.py:222
↓ 2 callersMethodaction_to_string
Convert an action number to a string representing the action. Args: action_number: an integer from the action space.
modelbased-rl/SampledMuZero/games/cartpole.py:186
↓ 2 callersMethodadd_phase
Add a phase to the trainer protocol. The score tensor can either be a scalar or vector, to support single and batched computations. Args
modelbased-rl/PlaNet/planet/training/trainer.py:118
↓ 2 callersMethodadversaries
(self, world)
multiagent-rl/easy-marl/envs/continuous_mpe/multiagent/scenarios/simple_world_comm.py:138
↓ 2 callersMethodadversaries
(self, world)
multiagent-rl/easy-marl/envs/continuous_mpe/multiagent/scenarios/simple_tag.py:80
← previousnext →301–400 of 3,230, ranked by callers