Functions — github.com/LXXXXR/CACOM

CodeHub MCPcopy Create free account

hub / github.com/LXXXXR/CACOM / functions

Functions226 in github.com/LXXXXR/CACOM

⨍ Functions 226◇ Types & classes 33

↓ 52 callersMethodlog_stat

(self, key, value, t)

src/utils/logging.py:16

↓ 20 callersMethodload

(self, path_name)

src/components/episode_buffer.py:266

↓ 17 callersMethodsave

(self, path_name)

src/components/episode_buffer.py:288

↓ 13 callersMethodinit_hidden

src/modules/agents/rnn_agent.py:15

↓ 13 callersMethodupdate

(self, data, bs=slice(None), ts=slice(None), mark_filled=True)

src/components/episode_buffer.py:88

↓ 10 callersMethodforward

(self, agent_qs, states)

src/modules/mixers/qmix.py:41

↓ 10 callersMethodparameters

src/controllers/basic_controller.py:59

↓ 9 callersMethodstep

Returns reward, terminated, info.

src/envs/join1.py:75

↓ 6 callersMethodsample

(self, batch_size)

src/components/episode_buffer.py:237

↓ 5 callersMethodget_obs

Returns all agent observations in a list.

src/envs/join1.py:124

↓ 5 callersMethodget_state

Returns the global state.

src/envs/join1.py:136

↓ 5 callersMethodto

src/components/episode_buffer.py:81

↓ 4 callersMethodafter_comm

(self, feat, response, hidden_state, bs)

src/modules/agents/cacom_agent.py:196

↓ 4 callersMethodcan_sample

(self, batch_size)

src/components/episode_buffer.py:234

↓ 4 callersMethodeval

src/components/epsilon_schedules.py:21

↓ 4 callersMethodget_avail_actions

Returns the available actions of all agents in a list.

src/envs/join1.py:144

↓ 4 callersMethodload_state

(self, other_mac)

src/controllers/basic_controller.py:62

↓ 4 callersMethodmax_t_filled

src/components/episode_buffer.py:198

↓ 4 callersFunctiontime_str

Convert seconds to a nicer string showing days, hours, minutes and seconds

src/utils/timehelper.py:28

↓ 4 callersMethodtrain

(self, batch: EpisodeBatch, t_env: int, episode_num: int, show_demo=False, save_data=None)

src/learners/q_learner.py:38

↓ 3 callersFunction_get_config

(params, arg_name, subfolder)

↓ 3 callersMethodclose

src/envs/join1.py:168

↓ 3 callersMethodcuda

src/learners/q_learner.py:150

↓ 3 callersFunctionrecursive_dict_update

↓ 3 callersMethodrun

(self, test_mode=False)

src/runners/episode_runner.py:48

↓ 3 callersMethodselect_actions

(self, ep_batch, t_ep, t_env, bs=slice(None), test_mode=False)

src/controllers/basic_controller.py:19

↓ 2 callersMethod_build_inputs

(self, batch, t)

src/controllers/cacom_controller.py:156

↓ 2 callersMethod_get_num_items

(self, indexing_item, max_size)

src/components/episode_buffer.py:162

↓ 2 callersMethod_log

(self, returns, stats, prefix)

src/runners/parallel_runner.py:202

↓ 2 callersMethod_log

(self, returns, stats, prefix)

src/runners/episode_runner.py:115

↓ 2 callersMethod_new_data_sn

src/components/episode_buffer.py:169

↓ 2 callersMethod_parse_slices

src/components/episode_buffer.py:175

↓ 2 callersMethod_setup_data

(self, scheme, groups, batch_size, max_seq_length, preprocess)

src/components/episode_buffer.py:31

↓ 2 callersMethodclose_env

src/runners/episode_runner.py:40

↓ 2 callersMethodcomm_response

(self, feat, requests)

src/modules/agents/cacom_agent.py:176

↓ 2 callersMethodget_obs_size

Returns the shape of the observation

src/envs/lbforaging/foraging.py:98

↓ 2 callersMethodinsert_episode_batch

(self, ep_batch)

src/components/episode_buffer.py:217

↓ 2 callersMethodpre_comm

(self, inputs, hidden_state, bs)

src/modules/agents/cacom_agent.py:138

↓ 2 callersMethodreset

src/runners/parallel_runner.py:61

↓ 2 callersFunctionsave_one_buffer

(args, save_buffer, env_name, from_start=False)

↓ 2 callersMethodseed

src/envs/join1.py:171

↓ 2 callersMethodseed

src/envs/lbforaging/foraging.py:141

↓ 2 callersMethodselect_action

(self, agent_inputs, avail_actions, t_env, test_mode=False)

src/components/action_selectors.py:18

↓ 1 callersMethod__init__

src/modules/agents/cacom_agent.py:334

↓ 1 callersMethod__init__

(self, bit=8, symmetric=True, per_channel=False)

src/modules/agents/rnn_agent.py:45

↓ 1 callersMethod__init__

(self, scheme, groups, batch_size, max_seq

src/components/episode_buffer.py:8

↓ 1 callersMethod_build_agents

(self, input_shape)

src/controllers/basic_controller.py:74

↓ 1 callersMethod_build_agents

(self, input_shape)

src/controllers/cacom_controller.py:153

↓ 1 callersMethod_build_inputs

(self, batch, t)

src/controllers/basic_controller.py:77

↓ 1 callersMethod_build_inputs

(self, batch, t=None)

src/modules/critics/coma.py:29

↓ 1 callersMethod_check_safe_view

(self, v, dest)

src/components/episode_buffer.py:115

↓ 1 callersFunction_get_comment

(params, arg_name)

↓ 1 callersMethod_get_input_shape

src/controllers/basic_controller.py:94

↓ 1 callersMethod_get_input_shape

src/controllers/cacom_controller.py:177

↓ 1 callersMethod_get_input_shape

src/modules/critics/coma.py:61

↓ 1 callersMethod_log_for_loss

(self, losses: dict, t)

src/learners/cacom_learner.py:229

↓ 1 callersMethod_process_loss

(self, losses: list, batch: EpisodeBatch)

src/learners/cacom_learner.py:216

↓ 1 callersMethod_train_critic

(self, batch, rewards, terminated, actions, avail_actions, mask, bs, max_t)

src/learners/coma_learner.py:100

↓ 1 callersMethod_update_targets

src/learners/qtran_learner.py:183

↓ 1 callersMethod_update_targets

src/learners/q_learner.py:144

↓ 1 callersMethod_update_targets

src/learners/dmaq_qatten_learner.py:196

↓ 1 callersMethod_update_targets

src/learners/coma_learner.py:150

↓ 1 callersMethod_update_targets

src/learners/cacom_learner.py:233

↓ 1 callersFunctionargs_sanity_check

↓ 1 callersFunctionbuild_td_lambda_targets

(rewards, terminated, mask, target_qs, n_agents, gamma, td_lambda)

src/utils/rl_utils.py:4

↓ 1 callersMethodcal_gate_labels

(self, inputs, hidden_state, bs, exp_gate)

src/modules/agents/cacom_agent.py:226

↓ 1 callersMethodcalc

(self, agent_qs, states, actions=None, max_q_i=None, is_v=False)

src/modules/mixers/dmaq_general.py:53

↓ 1 callersMethodcalc

(self, agent_qs, states, actions=None, max_q_i=None, is_v=False)

src/modules/mixers/dmaq_qatten.py:46

↓ 1 callersMethodcalc_adv

(self, agent_qs, states, actions, max_q_i)

src/modules/mixers/dmaq_general.py:36

↓ 1 callersMethodcalc_adv

(self, agent_qs, states, actions, max_q_i)

src/modules/mixers/dmaq_qatten.py:29

↓ 1 callersMethodcalc_v

(self, agent_qs)

src/modules/mixers/dmaq_general.py:31

↓ 1 callersMethodcalc_v

(self, agent_qs)

src/modules/mixers/dmaq_qatten.py:24

↓ 1 callersMethodcalculate_aux_loss

_summary_ Args: h (_type_): _description_ bs (_type_): _description_ feat (_type_): _description_

src/modules/agents/cacom_agent.py:289

↓ 1 callersFunctionconfig_copy

↓ 1 callersMethodcuda

src/learners/dmaq_qatten_learner.py:202

↓ 1 callersFunctionevaluate_sequential

↓ 1 callersMethodextend

(self, scheme, groups=None)

src/components/episode_buffer.py:78

↓ 1 callersMethodforward

(self, ep_batch, t, test_mode=False)

src/controllers/basic_controller.py:26

↓ 1 callersMethodforward

(self, ep_batch, t, all_through=False, test_mode=False, **kwargs)

src/controllers/cacom_controller.py:33

↓ 1 callersMethodforward_gate

(self, ep_batch, t)

src/controllers/cacom_controller.py:81

↓ 1 callersMethodgate_parameters

src/controllers/cacom_controller.py:130

↓ 1 callersMethodget_avail_agent_actions

Returns the available actions for agent_id.

src/envs/join1.py:148

↓ 1 callersMethodget_avail_agent_actions

Returns the available actions for agent_id

src/envs/lbforaging/foraging.py:115

↓ 1 callersMethodget_env_info

src/runners/parallel_runner.py:51

↓ 1 callersMethodget_env_info

src/envs/join1.py:178

↓ 1 callersFunctionget_logger

(name, log_file=None, log_level=logging.INFO)

src/utils/logging.py:45

↓ 1 callersMethodget_obs

Returns all agent observations in a list

src/envs/lbforaging/foraging.py:90

↓ 1 callersMethodget_obs_agent

Returns observation for agent_id.

src/envs/join1.py:128

↓ 1 callersMethodget_obs_size

Returns the size of the observation.

src/envs/join1.py:132

↓ 1 callersMethodget_obs_size

Returns the shape of the observation

src/envs/multiagentenv.py:15

↓ 1 callersMethodget_state

src/envs/lbforaging/foraging.py:102

↓ 1 callersMethodget_state_size

Returns the size of the global state.

src/envs/join1.py:140

↓ 1 callersMethodget_state_size

Returns the shape of the state

src/envs/multiagentenv.py:22

↓ 1 callersMethodget_state_size

Returns the shape of the state

src/envs/lbforaging/foraging.py:108

↓ 1 callersMethodget_stats

src/envs/join1.py:186

↓ 1 callersMethodget_total_actions

Returns the total number of actions an agent could ever take.

src/envs/join1.py:152

↓ 1 callersMethodget_total_actions

Returns the total number of actions an agent could ever take

src/envs/multiagentenv.py:33

↓ 1 callersMethodget_total_actions

Returns the total number of actions an agent could ever take

src/envs/lbforaging/foraging.py:123

↓ 1 callersFunctiongrad_scale

src/modules/agents/rnn_agent.py:27

↓ 1 callersMethodinfer_output_info

(self, vshape_in, dtype_in)

src/components/transforms.py:21

next →1–100 of 226, ranked by callers