docstr: Given input observations, use the model to make predictions (e.g. action or value).
  
  :param inputs: current state (i.e. observations, measurements, goals, etc.)
      (e.g. `{'observation': numpy.ndarray}` of shape (batch_size, observation_space_size))
  :param outputs: list of outputs to return. Return all outputs if unspecified. Type of the list elements
      depends on the framework backend.
  :param squeeze_output: call squeeze_list on output before returning if True
  :param initial_feed_dict: a dictionary of extra inputs for forward pass.
  :return: predictions of action or value of shape (batch_size, action_space_size) for action predictions)
docstr_tok: ["Given","input","observations",",","use","the","model","to","make","predictions","(","e.g",".","action","or","value",")",".",":","param","inputs",":","current","state","(","i.e",".","observations",",","measurements",",","goals",",","etc",".",")","(","e.g",".","`","{","'observation","'",":","numpy.ndarray","}","`","of","shape","(","batch_size",",","observation_space_size",")",")",":","param","outputs",":","list","of","outputs","to","return",".","Return","all","outputs","if","unspecified",".","Type","of","the","list","elements","depends","on","the","framework","backend",".",":","param","squeeze_output",":","call","squeeze_list","on","output","before","returning","if","True",":","param","initial_feed_dict",":","a","dictionary","of","extra","inputs","for","forward","pass",".",":","return",":","predictions","of","action","or","value","of","shape","(","batch_size",",","action_space_size",")","for","action","predictions",")"]
code: def predict(
      self,
      inputs: Dict[str, np.ndarray],
      outputs: List[Any] = None,
      squeeze_output: bool = True,
      initial_feed_dict: Dict[Any, np.ndarray] = None,
  ) -> Tuple[np.ndarray, ...]:
      """
      Given input observations, use the model to make predictions (e.g. action or value).
  
      :param inputs: current state (i.e. observations, measurements, goals, etc.)
          (e.g. `{'observation': numpy.ndarray}` of shape (batch_size, observation_space_size))
      :param outputs: list of outputs to return. Return all outputs if unspecified. Type of the list elements
          depends on the framework backend.
      :param squeeze_output: call squeeze_list on output before returning if True
      :param initial_feed_dict: a dictionary of extra inputs for forward pass.
      :return: predictions of action or value of shape (batch_size, action_space_size) for action predictions)
      """
      raise NotImplementedError
  
imports: from typing import Any, Dict, List, Tuple
  import numpy
  from rl_coach.base_parameters import AgentParameters
  from rl_coach.saver import SaverCollection
  from rl_coach.spaces import SpacesDefinition
label_desc: Use image node_id 3 for calling the Architecture obj's underlying member method code with example usage: obj.predict(inputs, outputs, squeeze_output, initial_feed_dict) without return types