MCPcopy Create free account

hub / github.com/casterkay/Energy-Efficient-5G-RL / functions

Functions683 in github.com/casterkay/Energy-Efficient-5G-RL

↓ 97 callersMethodadd_argument
(self, *args, **kwargs)
agents/option-critic/utils/config.py:81
↓ 46 callersMethodmean
(self)
agents/option-critic/utils/torch_utils.py:200
↓ 29 callersMethodsize
(self)
agents/option-critic/component/replay.py:142
↓ 26 callersFunctionlayer_init
(layer, w_scale=1.0)
agents/option-critic/network/network_utils.py:23
↓ 25 callersMethodclose
(self)
env/env_wrappers.py:97
↓ 25 callersFunctionnotice
(msg, *args, **kwds)
utils.py:46
↓ 20 callersFunction_flatten
(T, N, x)
trainers/utils/shared_buffer.py:6
↓ 20 callersFunction_flatten
(T, N, x)
trainers/utils/separated_buffer.py:7
↓ 20 callersMethodreset
Reset all the environments and return an array of observations, or a dict of observation arrays. If step_async is still doin
env/env_wrappers.py:53
↓ 19 callersFunctiontensor
(x)
agents/option-critic/utils/torch_utils.py:20
↓ 18 callersFunctioncheck
(input)
trainers/utils/__init__.py:10
↓ 17 callersMethodrender
(self, mode='human')
env/env_wrappers.py:114
↓ 15 callersMethoddenormalize
Transform normalized data back into original distribution
trainers/utils/valuenorm.py:68
↓ 15 callersMethodstep
Step the environments synchronously. This is available for backwards compatibility.
env/env_wrappers.py:105
↓ 14 callersMethodadd
(self, op)
agents/option-critic/utils/torch_utils.py:136
↓ 11 callersFunctiondiv0
Replace 0 with eps in y before dividing x by y.
utils.py:102
↓ 10 callersMethod__init__
(self, num_envs, observation_space, cent_observation_space, action_space)
env/env_wrappers.py:46
↓ 10 callersFunction_cast
(x)
trainers/utils/shared_buffer.py:10
↓ 10 callersFunction_cast
(x)
trainers/utils/separated_buffer.py:11
↓ 10 callersMethodload_state_dict
(self, _)
agents/option-critic/utils/normalizer.py:24
↓ 10 callersMethodupdate
(self, input_vector)
agents/mappo/nn/popart.py:49
↓ 9 callersMethod__init__
(self)
analysis/keras-model.py:11
↓ 9 callersMethod__init__
(self, action_dim, actor_body_fn, critic_body_fn,
agents/option-critic/network/network_heads.py:259
↓ 9 callersMethodentropy
(self)
agents/mappo/nn/distributions.py:36
↓ 9 callersMethodmul
(self, coef)
agents/option-critic/utils/torch_utils.py:146
↓ 9 callersMethodstate_dict
(self)
agents/option-critic/utils/normalizer.py:21
↓ 8 callersFunctionbox_env_ndims
(env)
network/env_utils.py:22
↓ 8 callersMethodget_bs
(self, id)
network/network.py:187
↓ 8 callersMethodreset
(self)
agents/option-critic/component/envs.py:183
↓ 7 callersMethodget
(self, s)
agents/option-critic/utils/sum_tree.py:63
↓ 7 callersMethodsample
(self, batch_size=None)
agents/option-critic/component/replay.py:92
↓ 6 callersMethodclone
(self)
agents/option-critic/utils/torch_utils.py:159
↓ 6 callersMethodgather
Gather Q-values for given input and actions. :param x: (torch.Tensor) input to network. :param actions: (torch.Tensor) action
agents/dqn.py:49
↓ 6 callersFunctioninit
(module, weight_init, bias_init, gain=1)
agents/mappo/nn/util.py:8
↓ 6 callersMethodlog_prob
(self, action)
agents/option-critic/utils/torch_utils.py:95
↓ 6 callersMethodlog_probs
(self, actions)
agents/mappo/nn/distributions.py:33
↓ 6 callersFunctionth2np
Convert torch tensor to a numpy array.
trainers/utils/__init__.py:15
↓ 5 callersMethodadd_scalar
(self, tag, value, step=None, log_level=0)
agents/option-critic/utils/logger.py:55
↓ 5 callersMethodclose
(self)
agents/option-critic/agent/BaseAgent.py:24
↓ 5 callersMethodconsume_energy
(self, e, k)
network/base_station.py:506
↓ 5 callersMethodinfo_dict
(self)
env/environment.py:232
↓ 5 callersMethodreset_task
(self)
env/env_wrappers.py:264
↓ 5 callersMethodseed
(cls, seed)
traffic/model.py:62
↓ 4 callersMethod__init__
(self, name, num_envs=1, single_process=True,
agents/option-critic/component/envs.py:154
↓ 4 callersMethodannotate_obs
(cls, obs)
network/network.py:328
↓ 4 callersFunctionconcat_box_envs
(*envs)
network/env_utils.py:14
↓ 4 callersMethoddebiased_mean_var
(self)
agents/mappo/nn/popart.py:72
↓ 4 callersMethodentropy
(self)
agents/option-critic/utils/torch_utils.py:98
↓ 4 callersMethodfeed
(self, data)
agents/option-critic/component/replay.py:32
↓ 4 callersMethodfilter_log_dirs
(self, pattern, negative_pattern=' ', root='./log', **kwargs)
agents/option-critic/utils/plot.py:68
↓ 4 callersFunctionget_shape_from_obs_space
(obs_space)
trainers/utils/space_utils.py:1
↓ 4 callersFunctionkwds_str
(**kwds)
utils.py:69
↓ 4 callersMethodload_log_dirs
(self, dirs, **kwargs)
agents/option-critic/utils/plot.py:85
↓ 4 callersFunctionmake_box_env
(bounds, dtype=np.float32)
network/env_utils.py:6
↓ 4 callersMethodsample
(self)
agents/mappo/nn/distributions.py:15
↓ 4 callersMethodstep
(self, actions)
agents/option-critic/component/envs.py:186
↓ 4 callersMethodupdate_data_rate
(self)
network/user_equipment.py:168
↓ 3 callersMethod__init__
(self, bias)
agents/mappo/nn/distributions.py:108
↓ 3 callersMethod__init__
(self, read_only=False)
agents/option-critic/utils/normalizer.py:12
↓ 3 callersMethod__init__
(self, memory_size, keys=None)
agents/option-critic/component/replay.py:21
↓ 3 callersMethod__init__
(self, state_dim, hidden_units=(64, 64), gate=F.relu, noisy_linear=False)
agents/option-critic/network/network_bodies.py:51
↓ 3 callersMethod_sample
(self)
agents/option-critic/agent/BaseAgent.py:115
↓ 3 callersMethodact
(self, obs, **kwargs)
agents/base.py:7
↓ 3 callersMethodactor
(self, phi)
agents/option-critic/network/network_heads.py:166
↓ 3 callersMethodadd_stat
(self, key, val)
network/network.py:339
↓ 3 callersMethodclose
(self)
trainers/trainer.py:73
↓ 3 callersMethodcritic
(self, phi, a)
agents/option-critic/network/network_heads.py:169
↓ 3 callersFunctiondB2lin
(dB)
utils.py:63
↓ 3 callersMethoddisconnect
(self)
network/user_equipment.py:195
↓ 3 callersFunctionget_mappo_config
Network parameters: --share_policy by default True, all agents will share the same network; set to make training agents use d
trainers/mappo_trainer.py:12
↓ 3 callersFunctionget_time_str
()
agents/option-critic/utils/misc.py:38
↓ 3 callersMethodmode
(self)
agents/mappo/nn/distributions.py:39
↓ 3 callersMethodpop_from_queue
(self, ue=None)
network/base_station.py:362
↓ 3 callersMethodprep_rollout
(self)
agents/mappo/policy.py:119
↓ 3 callersMethodprint_info
(self)
traffic/model.py:89
↓ 3 callersFunctionrefactor
(df)
analysis/benchmark.py:50
↓ 3 callersFunctionrender_bss
(net, frame, show_id=False)
visualize/render.py:95
↓ 3 callersMethodsample
(self, step)
trainers/mappo_trainer.py:506
↓ 3 callersMethodseed
(self, seed=None)
env/environment.py:128
↓ 3 callersMethodswitch_sleep_mode
(self, mode)
network/base_station.py:249
↓ 3 callersMethodtrain
Collect training data, perform training updates, and evaluate policy.
trainers/trainer.py:46
↓ 3 callersMethodtransform_noise
(self, x)
agents/option-critic/network/network_utils.py:82
↓ 3 callersMethodupdate
(self, idx, p)
agents/option-critic/utils/sum_tree.py:54
↓ 3 callersMethodupdate_power_allocation
(self)
network/base_station.py:375
↓ 2 callersMethod__array__
(self, dtype=None)
agents/option-critic/component/envs.py:103
↓ 2 callersMethod_get_time_loc
(self, time)
traffic/model.py:107
↓ 2 callersMethod_set_up
(self)
agents/option-critic/agent/BaseAgent.py:148
↓ 2 callersMethodadd_to_queue
(self, ue)
network/base_station.py:354
↓ 2 callersFunctioncompute_channel_gain
(distances, C=dB2lin(-const), a=alpha)
network/channel.py:15
↓ 2 callersMethodcompute_data_rate
Returns: The data_rate of the UE in bits/s.
network/user_equipment.py:159
↓ 2 callersMethodconnect
(self, ue)
network/base_station.py:303
↓ 2 callersMethodconstruct_transition
(self, index)
agents/option-critic/component/replay.py:112
↓ 2 callersFunctionduplicate_box_env
(env, n)
network/env_utils.py:18
↓ 2 callersMethodget_all_ue_stats
(self)
network/base_station.py:576
↓ 2 callersMethodget_arrival_rates
(self, time, dt)
traffic/model.py:120
↓ 2 callersMethodget_cent_obs
(self)
env/environment.py:160
↓ 2 callersFunctionget_config
The configuration parser for common hyperparameters of all environment. Please reach each `scripts/train/<env>_runner.py` file to find priva
arguments.py:5
↓ 2 callersFunctionget_env_config
()
arguments.py:145
↓ 2 callersFunctionget_env_fn
(rank)
train.py:40
↓ 2 callersFunctionget_env_kwargs
(args)
train.py:17
next →1–100 of 683, ranked by callers