Fix learningRate param in snake-dqn (#353)

Author: pskucherov

snake-dqn/agent.js

@@ -37,6 +37,7 @@ export class SnakeGameAgent {
    *   - `epsilonDecayFrames` {number} The # of frames over which the value of
    *     `epsilon` decreases from `episloInit` to `epsilonFinal`, via a linear
    *     schedule.
+   *   - `learningRate` {number} The learning rate to use during training.
    */
   constructor(game, config) {
     assertPositiveInteger(config.epsilonDecayFrames);

snake-dqn/train.js

@@ -253,7 +253,8 @@ async function main() {
     replayBufferSize: args.replayBufferSize,
     epsilonInit: args.epsilonInit,
     epsilonFinal: args.epsilonFinal,
-    epsilonDecayFrames: args.epsilonDecayFrames
+    epsilonDecayFrames: args.epsilonDecayFrames,
+    learningRate: args.learningRate
   });
 
   await train(