Method predict

rl2/atari/dqn_tf.py:243–244

(self, states)

    self.session = session

  def predict(self, states):
    return self.session.run(self.predict_op, feed_dict={self.X: states})

  def update(self, states, actions, targets):
    c, _ = self.session.run(