class_docstr: Sample Rainbow agent to visualize Q-values and rewards.
class_docstr_tok: ["Sample","Rainbow","agent","to","visualize","Q-values","and","rewards","."]
code: def get_probabilities(self):
      return self._sess.run(
          tf.squeeze(self._net_outputs.probabilities),
          {self.state_ph: self.state},
      )
  
imports: from __future__ import absolute_import
  from __future__ import division
  from __future__ import print_function
  import os
  from absl import logging
  from dopamine.agents.dqn import dqn_agent
  from dopamine.agents.rainbow import rainbow_agent
  from dopamine.discrete_domains import atari_lib
  from dopamine.discrete_domains import iteration_statistics
  from dopamine.discrete_domains import run_experiment
  from dopamine.utils import agent_visualizer
  from dopamine.utils import atari_plotter
  from dopamine.utils import bar_plotter
  from dopamine.utils import line_plotter
  import gin
  import numpy
  import tensorflow
  import tf_slim
label_desc: Use image node_id 4 for calling the MyRainbowAgent obj's underlying member method code with example usage: obj.get_probabilities() and returns: self