a91082900/RecSim_FlashcardLearning
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

change initial weights for ucb agent

Browse Source
This commit is contained in:
Jerry Wu 2023-10-29 17:37:32 +08:00
parent fe56de9d76
commit 003ef8effe
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
agent/ucb_agent.py
View File

@ -6,7 +6,7 @@ class UCBAgent(AbstractEpisodicRecommenderAgent):
def __init__(self, sess, observation_space, action_space, eval_mode, alpha=1.0, learning_rate=0.001, summary_writer=None): def __init__(self, sess, observation_space, action_space, eval_mode, alpha=1.0, learning_rate=0.001, summary_writer=None):
super(UCBAgent, self).__init__(action_space, summary_writer) super(UCBAgent, self).__init__(action_space, summary_writer)
self._num_candidates = int(action_space.nvec[0]) self._num_candidates = int(action_space.nvec[0])
self._W = tf.Variable(np.random.normal(size=(self._num_candidates, 3)), name='W') self._W = tf.Variable(np.random.uniform(0, 10, size=(self._num_candidates, 3)), name='W')
self._sess = sess self._sess = sess
self._return_idx = None self._return_idx = None
self._prev_pred_pr = None self._prev_pred_pr = None