distributional rl

gzz1996 · Dec 21, 2018 · 772c0dd · 772c0dd
1 parent 7d03dd0
commit 772c0dd
Show file tree

Hide file tree

Showing 2 changed files with 31 additions and 31 deletions.
diff --git a/.idea/workspace.xml b/.idea/workspace.xml
diff --git a/RL/Basic-DisRL-Demo/Categorical_DQN.py b/RL/Basic-DisRL-Demo/Categorical_DQN.py
@@ -60,7 +60,7 @@ def build_layers(self, state, action, c_names, units_1, units_2, w_i, b_i, reg=N
             concatenated = tf.concat([dense2, tf.cast(action, tf.float32)], 1)
         with tf.variable_scope('dense3'):
             dense3 = dense(concatenated, self.atoms, [self.atoms], w_i, b_i) # 返回
-        return dense3
+        return tf.nn.softmax(dense3)
 
     def build_cate_dqn_net(self):
         with tf.variable_scope('target_net'):