补充softmax激活函数

Author: Sharpiless

train_keras.py

@@ -30,6 +30,7 @@ def build_network(num_actions):
     model.add(Dense(512))
     model.add(Activation('relu'))
     model.add(Dense(num_actions))
+    model.add(Activation('softmax'))
 
     if os.path.exists("dqn.h5"):
         print("Loading weights from dqn.h5 .....")
@@ -119,4 +120,4 @@ def train_network():
 if __name__ == "__main__":
 
     train_network()
-    
+