MCPcopy Create free account

hub / github.com/HCPLab-SYSU/Medical_DS / functions

Functions195 in github.com/HCPLab-SYSU/Medical_DS

↓ 24 callersMethodsample_noise
(self)
qlearning/dqn_prior.py:122
↓ 20 callersMethodfeature_size
(self, input_shape)
qlearning/dqn_prior.py:410
↓ 9 callersMethod__init__
(self, input_shape, num_actions, noisy=False, sigma_init=0.5, body=SimpleBody)
qlearning/dqn_prior.py:104
↓ 8 callersFunctiontext_to_dict
Read in a text file as a dictionary where keys are text and values are indices (line numbers)
utils/utils.py:11
↓ 8 callersMethodupdate
Update the state based on the latest action
dialog_system/state_tracker.py:57
↓ 6 callersFunctionload_pickle
(path)
utils/utils.py:6
↓ 5 callersMethodinitialize_episode
(self)
agents/agent.py:91
↓ 5 callersMethodnext_turn
This function initiates each subsequent exchange between agent and user (agent first)
dialog_system/dialog_manager.py:42
↓ 3 callersMethodclear
(self)
utils/ReplayMemory.py:37
↓ 3 callersMethodget_sym_flag
(self, batch_state)
agents/agent.py:206
↓ 3 callersMethodlayer_init
(self, module, weight_init, bias_init, gain=1)
qlearning/dqn_prior.py:413
↓ 3 callersMethodprepare_state_representation
(self, state)
agents/agent.py:122
↓ 3 callersMethodprepare_state_representation
Create the representation for each state
agents/agent_dqn_prior.py:91
↓ 3 callersMethodprint_function
Print Function
dialog_system/dialog_manager.py:85
↓ 3 callersMethodpush
(self, transition)
utils/ReplayMemory.py:16
↓ 3 callersMethodsample
(self, batch_size)
utils/ReplayMemory.py:21
↓ 3 callersFunctionsave_model
(path, agt, agent, cur_epoch, best_epoch=0, best_success_rate=0.0, best_ave_turns=0.0, tr_success_rate=0.0, te
train.py:180
↓ 3 callersMethodsum
Returns arr[start] + ... + arr[end]
utils/data_structures.py:94
↓ 3 callersMethodtrain
(self)
agents/agent.py:254
↓ 2 callersMethod__init__
Build a Segment Tree data structure. https://en.wikipedia.org/wiki/Segment_tree Can be used as regular array, but with two imp
utils/data_structures.py:5
↓ 2 callersMethod_scale_noise
(self, size)
qlearning/layers.py:30
↓ 2 callersMethodaction_index
Return the index of action
agents/agent_dqn_prior.py:420
↓ 2 callersMethoddisease_from_dict
(self, current_slots)
agents/agent_dqn_prior.py:276
↓ 2 callersMethoddisease_from_dict_rate
(self, current_slots)
agents/agent_dqn_prior.py:286
↓ 2 callersMethodget_state_for_agent
Get the state representatons to send to agent
dialog_system/state_tracker.py:50
↓ 2 callersMethodinitialize_episode
Initialize a new episode (dialog) state['history_slots']: keeps all the informed_slots state['rest_slots']: keep all the slots (which
usersim/usersim_rule.py:43
↓ 2 callersMethodinsert
(self, current_obs, action, action_log_prob, value_pred, reward, mask)
utils/RolloutStorage.py:19
↓ 2 callersFunctionload_data
(indir, smooth, bin_size)
utils/plot.py:46
↓ 2 callersMethodmin
Returns min(arr[start], ..., arr[end])
utils/data_structures.py:136
↓ 2 callersMethodpredict
(self, x)
qlearning/dqn_prior.py:128
↓ 2 callersMethodreduce
Returns result of applying `self.operation` to a contiguous subsequence of the array. self.operation(arr[start], operation(arr[sta
utils/data_structures.py:47
↓ 2 callersMethodsingle_batch
(self)
agents/agent.py:240
↓ 2 callersFunctiontest
(test_size, dialog_manager, output=False)
predict.py:153
↓ 2 callersFunctiontest
(simu_size, data_split, out=False)
train.py:255
↓ 1 callersMethod__init__
(self, env=None)
utils/wrappers.py:28
↓ 1 callersMethod__init__
(self, input_shape, num_actions, noisy=False, sigma_init=0.5)
qlearning/network_bodies.py:11
↓ 1 callersMethod_reduce_helper
(self, start, end, node, node_start, node_end)
utils/data_structures.py:32
↓ 1 callersMethod_sample_goal
sample a user goal
usersim/usersim_test.py:77
↓ 1 callersMethod_sample_goal
sample a user goal
usersim/usersim_rule.py:89
↓ 1 callersMethodaction_index
Return the index of action
agents/agent.py:382
↓ 1 callersMethodbeta_by_frame
(self, frame_idx)
utils/ReplayMemory.py:56
↓ 1 callersMethodbeta_by_frame
(self, frame_idx)
utils/ReplayMemory.py:111
↓ 1 callersMethodcompute_loss
(self, batch_vars)
agents/agent.py:213
↓ 1 callersMethodcorrupt
Randomly corrupt an action with error probs (slot_err_probability and slot_err_mode) on Slot and Intent (intent_err_probability).
usersim/usersim_test.py:84
↓ 1 callersMethodcorrupt
Randomly corrupt an action with error probs (slot_err_probability and slot_err_mode) on Slot and Intent (intent_err_probability).
usersim/usersim_rule.py:95
↓ 1 callersMethoddeclare_memory
(self)
agents/agent.py:116
↓ 1 callersMethoddeclare_networks
(self, path)
agents/agent.py:94
↓ 1 callersMethoddetect_repeat
(self, state, act_ind)
agents/agent.py:284
↓ 1 callersMethoddialog_history_dictionaries
Return the dictionary representation of the dialog history (includes values)
dialog_system/state_tracker.py:46
↓ 1 callersMethoddisease_from_dict
(self, current_slots, sym_flag)
agents/agent.py:295
↓ 1 callersFunctionfix_point
(x, y, interval)
utils/plot.py:22
↓ 1 callersMethodforward
(self, state, sym_flag)
qlearning/dqn_prior.py:81
↓ 1 callersMethodforward
(self, x)
qlearning/dqn_prior.py:115
↓ 1 callersMethodinit_hidden
(self, batch_size)
qlearning/dqn_prior.py:362
↓ 1 callersMethodinitialize_episode
Initialize a new episode (dialog), flush the current state and tracked slots
dialog_system/state_tracker.py:33
↓ 1 callersMethodload_specific_state_dict
(self,path)
agents/agent.py:109
↓ 1 callersMethodload_trained_DQN
Load the trained DQN from a file
agents/agent_dqn_prior.py:477
↓ 1 callersFunctionmain
(params)
usersim/usersim_test.py:256
↓ 1 callersFunctionmain
(params)
usersim/usersim_rule.py:267
↓ 1 callersMethodnext
(self, system_action)
usersim/usersim.py:30
↓ 1 callersMethodprep_minibatch
(self)
agents/agent.py:187
↓ 1 callersMethodregister_experience_replay_tuple
(self, s_t, a_t, reward, s_tplus1, episode_over)
agents/agent.py:170
↓ 1 callersMethodreset_parameters
(self)
qlearning/layers.py:23
↓ 1 callersMethodresponse_inform
(self, system_action)
usersim/usersim_test.py:246
↓ 1 callersMethodresponse_inform
(self, system_action)
usersim/usersim_rule.py:257
↓ 1 callersMethodresponse_request
Response for Request (System Action)
usersim/usersim_test.py:184
↓ 1 callersMethodresponse_request
Response for Request (System Action)
usersim/usersim_rule.py:195
↓ 1 callersMethodresponse_thanks
Response for Thanks (System Action)
usersim/usersim_test.py:173
↓ 1 callersMethodresponse_thanks
Response for Thanks (System Action)
usersim/usersim_rule.py:184
↓ 1 callersMethodreward_function
Reward Function 1: a reward function based on the dialog_status
dialog_system/dialog_manager.py:74
↓ 1 callersMethodrule_policy
Rule Policy
agents/agent.py:325
↓ 1 callersMethodrule_policy
Rule Policy
agents/agent_dqn_prior.py:351
↓ 1 callersMethodrun_policy
(self, rep, state)
agents/agent.py:269
↓ 1 callersMethodrun_policy
epsilon-greedy policy
agents/agent_dqn_prior.py:256
↓ 1 callersMethodsample_noise
(self)
qlearning/layers.py:34
↓ 1 callersMethodsample_noise
(self)
qlearning/network_bodies.py:33
↓ 1 callersMethodsearch_batch_action_mask
(self, Xs)
agents/agent_dqn_prior.py:224
↓ 1 callersMethodsimu_state_to_action
(self, state, goal)
agents/agent_dqn_prior.py:65
↓ 1 callersFunctionsimulation_epoch
(simulation_epoch_size, output=False)
train.py:198
↓ 1 callersFunctionsmooth_reward_curve
(x, y)
utils/plot.py:11
↓ 1 callersMethodstart_action
(self)
usersim/usersim_test.py:61
↓ 1 callersMethodstart_action
(self)
usersim/usersim_rule.py:72
↓ 1 callersMethodstate_to_action
(self, state)
agents/agent.py:376
↓ 1 callersFunctiontraining
(count)
train.py:326
↓ 1 callersMethodupdate_priorities
(self, batch_indices, batch_priorities)
utils/ReplayMemory.py:153
↓ 1 callersMethodupdate_target_model
(self)
agents/agent.py:390
↓ 1 callersFunctionwarm_start_simulation
()
train.py:295
Method__getitem__
(self, idx)
utils/data_structures.py:81
Method__init__
(self, capacity, success_rate_threshold=0.3)
utils/ReplayMemory.py:10
Method__init__
(self, success_rate_threshold=0.3)
utils/ReplayMemory.py:28
Method__init__
(self, alpha=0.6, beta_start=0.4, beta_frames=100000)
utils/ReplayMemory.py:47
Method__init__
(self, capacity, alpha=0.6, beta_start=0.4, beta_frames=100000)
utils/ReplayMemory.py:101
Method__init__
(self, capacity, sequence_length=10)
utils/ReplayMemory.py:303
Method__init__
(self, capacity)
utils/data_structures.py:87
Method__init__
(self, capacity)
utils/data_structures.py:129
Method__init__
(self, num_steps, num_processes, obs_shape, action_space, device, USE_GAE=True, gae_tau=0.95)
utils/RolloutStorage.py:5
Method__init__
(self, env)
utils/wrappers.py:16
Method__init__
(self)
utils/hyperparameters.py:6
Method__init__
constructor for statetracker takes movie knowledge base and initializes a new episode Arguments: Class Variables: history_v
dialog_system/state_tracker.py:16
Method__init__
(self, agent, user, act_set, slot_set, params)
dialog_system/dialog_manager.py:15
next →1–100 of 195, ranked by callers