Code
Hub
Workspaces
Following
Trending
Connect
MCP
copy
Create free account
hub
/
github.com/casterkay/Energy-Efficient-5G-RL
/ functions
Functions
683 in github.com/casterkay/Energy-Efficient-5G-RL
⨍
Functions
683
◇
Types & classes
118
↳
Endpoints
2
↓ 97 callers
Method
add_argument
(self, *args, **kwargs)
agents/option-critic/utils/config.py:81
↓ 46 callers
Method
mean
(self)
agents/option-critic/utils/torch_utils.py:200
↓ 29 callers
Method
size
(self)
agents/option-critic/component/replay.py:142
↓ 26 callers
Function
layer_init
(layer, w_scale=1.0)
agents/option-critic/network/network_utils.py:23
↓ 25 callers
Method
close
(self)
env/env_wrappers.py:97
↓ 25 callers
Function
notice
(msg, *args, **kwds)
utils.py:46
↓ 20 callers
Function
_flatten
(T, N, x)
trainers/utils/shared_buffer.py:6
↓ 20 callers
Function
_flatten
(T, N, x)
trainers/utils/separated_buffer.py:7
↓ 20 callers
Method
reset
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 callers
Function
tensor
(x)
agents/option-critic/utils/torch_utils.py:20
↓ 18 callers
Function
check
(input)
trainers/utils/__init__.py:10
↓ 17 callers
Method
render
(self, mode='human')
env/env_wrappers.py:114
↓ 15 callers
Method
denormalize
Transform normalized data back into original distribution
trainers/utils/valuenorm.py:68
↓ 15 callers
Method
step
Step the environments synchronously. This is available for backwards compatibility.
env/env_wrappers.py:105
↓ 14 callers
Method
add
(self, op)
agents/option-critic/utils/torch_utils.py:136
↓ 11 callers
Function
div0
Replace 0 with eps in y before dividing x by y.
utils.py:102
↓ 10 callers
Method
__init__
(self, num_envs, observation_space, cent_observation_space, action_space)
env/env_wrappers.py:46
↓ 10 callers
Function
_cast
(x)
trainers/utils/shared_buffer.py:10
↓ 10 callers
Function
_cast
(x)
trainers/utils/separated_buffer.py:11
↓ 10 callers
Method
load_state_dict
(self, _)
agents/option-critic/utils/normalizer.py:24
↓ 10 callers
Method
update
(self, input_vector)
agents/mappo/nn/popart.py:49
↓ 9 callers
Method
__init__
(self)
analysis/keras-model.py:11
↓ 9 callers
Method
__init__
(self, action_dim, actor_body_fn, critic_body_fn,
agents/option-critic/network/network_heads.py:259
↓ 9 callers
Method
entropy
(self)
agents/mappo/nn/distributions.py:36
↓ 9 callers
Method
mul
(self, coef)
agents/option-critic/utils/torch_utils.py:146
↓ 9 callers
Method
state_dict
(self)
agents/option-critic/utils/normalizer.py:21
↓ 8 callers
Function
box_env_ndims
(env)
network/env_utils.py:22
↓ 8 callers
Method
get_bs
(self, id)
network/network.py:187
↓ 8 callers
Method
reset
(self)
agents/option-critic/component/envs.py:183
↓ 7 callers
Method
get
(self, s)
agents/option-critic/utils/sum_tree.py:63
↓ 7 callers
Method
sample
(self, batch_size=None)
agents/option-critic/component/replay.py:92
↓ 6 callers
Method
clone
(self)
agents/option-critic/utils/torch_utils.py:159
↓ 6 callers
Method
gather
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 callers
Function
init
(module, weight_init, bias_init, gain=1)
agents/mappo/nn/util.py:8
↓ 6 callers
Method
log_prob
(self, action)
agents/option-critic/utils/torch_utils.py:95
↓ 6 callers
Method
log_probs
(self, actions)
agents/mappo/nn/distributions.py:33
↓ 6 callers
Function
th2np
Convert torch tensor to a numpy array.
trainers/utils/__init__.py:15
↓ 5 callers
Method
add_scalar
(self, tag, value, step=None, log_level=0)
agents/option-critic/utils/logger.py:55
↓ 5 callers
Method
close
(self)
agents/option-critic/agent/BaseAgent.py:24
↓ 5 callers
Method
consume_energy
(self, e, k)
network/base_station.py:506
↓ 5 callers
Method
info_dict
(self)
env/environment.py:232
↓ 5 callers
Method
reset_task
(self)
env/env_wrappers.py:264
↓ 5 callers
Method
seed
(cls, seed)
traffic/model.py:62
↓ 4 callers
Method
__init__
(self, name, num_envs=1, single_process=True,
agents/option-critic/component/envs.py:154
↓ 4 callers
Method
annotate_obs
(cls, obs)
network/network.py:328
↓ 4 callers
Function
concat_box_envs
(*envs)
network/env_utils.py:14
↓ 4 callers
Method
debiased_mean_var
(self)
agents/mappo/nn/popart.py:72
↓ 4 callers
Method
entropy
(self)
agents/option-critic/utils/torch_utils.py:98
↓ 4 callers
Method
feed
(self, data)
agents/option-critic/component/replay.py:32
↓ 4 callers
Method
filter_log_dirs
(self, pattern, negative_pattern=' ', root='./log', **kwargs)
agents/option-critic/utils/plot.py:68
↓ 4 callers
Function
get_shape_from_obs_space
(obs_space)
trainers/utils/space_utils.py:1
↓ 4 callers
Function
kwds_str
(**kwds)
utils.py:69
↓ 4 callers
Method
load_log_dirs
(self, dirs, **kwargs)
agents/option-critic/utils/plot.py:85
↓ 4 callers
Function
make_box_env
(bounds, dtype=np.float32)
network/env_utils.py:6
↓ 4 callers
Method
sample
(self)
agents/mappo/nn/distributions.py:15
↓ 4 callers
Method
step
(self, actions)
agents/option-critic/component/envs.py:186
↓ 4 callers
Method
update_data_rate
(self)
network/user_equipment.py:168
↓ 3 callers
Method
__init__
(self, bias)
agents/mappo/nn/distributions.py:108
↓ 3 callers
Method
__init__
(self, read_only=False)
agents/option-critic/utils/normalizer.py:12
↓ 3 callers
Method
__init__
(self, memory_size, keys=None)
agents/option-critic/component/replay.py:21
↓ 3 callers
Method
__init__
(self, state_dim, hidden_units=(64, 64), gate=F.relu, noisy_linear=False)
agents/option-critic/network/network_bodies.py:51
↓ 3 callers
Method
_sample
(self)
agents/option-critic/agent/BaseAgent.py:115
↓ 3 callers
Method
act
(self, obs, **kwargs)
agents/base.py:7
↓ 3 callers
Method
actor
(self, phi)
agents/option-critic/network/network_heads.py:166
↓ 3 callers
Method
add_stat
(self, key, val)
network/network.py:339
↓ 3 callers
Method
close
(self)
trainers/trainer.py:73
↓ 3 callers
Method
critic
(self, phi, a)
agents/option-critic/network/network_heads.py:169
↓ 3 callers
Function
dB2lin
(dB)
utils.py:63
↓ 3 callers
Method
disconnect
(self)
network/user_equipment.py:195
↓ 3 callers
Function
get_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 callers
Function
get_time_str
()
agents/option-critic/utils/misc.py:38
↓ 3 callers
Method
mode
(self)
agents/mappo/nn/distributions.py:39
↓ 3 callers
Method
pop_from_queue
(self, ue=None)
network/base_station.py:362
↓ 3 callers
Method
prep_rollout
(self)
agents/mappo/policy.py:119
↓ 3 callers
Method
print_info
(self)
traffic/model.py:89
↓ 3 callers
Function
refactor
(df)
analysis/benchmark.py:50
↓ 3 callers
Function
render_bss
(net, frame, show_id=False)
visualize/render.py:95
↓ 3 callers
Method
sample
(self, step)
trainers/mappo_trainer.py:506
↓ 3 callers
Method
seed
(self, seed=None)
env/environment.py:128
↓ 3 callers
Method
switch_sleep_mode
(self, mode)
network/base_station.py:249
↓ 3 callers
Method
train
Collect training data, perform training updates, and evaluate policy.
trainers/trainer.py:46
↓ 3 callers
Method
transform_noise
(self, x)
agents/option-critic/network/network_utils.py:82
↓ 3 callers
Method
update
(self, idx, p)
agents/option-critic/utils/sum_tree.py:54
↓ 3 callers
Method
update_power_allocation
(self)
network/base_station.py:375
↓ 2 callers
Method
__array__
(self, dtype=None)
agents/option-critic/component/envs.py:103
↓ 2 callers
Method
_get_time_loc
(self, time)
traffic/model.py:107
↓ 2 callers
Method
_set_up
(self)
agents/option-critic/agent/BaseAgent.py:148
↓ 2 callers
Method
add_to_queue
(self, ue)
network/base_station.py:354
↓ 2 callers
Function
compute_channel_gain
(distances, C=dB2lin(-const), a=alpha)
network/channel.py:15
↓ 2 callers
Method
compute_data_rate
Returns: The data_rate of the UE in bits/s.
network/user_equipment.py:159
↓ 2 callers
Method
connect
(self, ue)
network/base_station.py:303
↓ 2 callers
Method
construct_transition
(self, index)
agents/option-critic/component/replay.py:112
↓ 2 callers
Function
duplicate_box_env
(env, n)
network/env_utils.py:18
↓ 2 callers
Method
get_all_ue_stats
(self)
network/base_station.py:576
↓ 2 callers
Method
get_arrival_rates
(self, time, dt)
traffic/model.py:120
↓ 2 callers
Method
get_cent_obs
(self)
env/environment.py:160
↓ 2 callers
Function
get_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 callers
Function
get_env_config
()
arguments.py:145
↓ 2 callers
Function
get_env_fn
(rank)
train.py:40
↓ 2 callers
Function
get_env_kwargs
(args)
train.py:17
next →
1–100 of 683, ranked by callers