class_docstr: Manages a TensorFlow model for training.
class_docstr_tok: ["Manages","a","TensorFlow","model","for","training","."]
docstr: Returns the state of the runner.
docstr_tok: ["Returns","the","state","of","the","runner","."]
code: def get_state(self):
      """Returns the state of the runner."""
      return {
          "epoch": self.epoch,
          "weights": self.model.get_weights(),
          "optimizer_weights": self.model.optimizer.get_weights(),
      }
  
imports: import logging
  import json
  import os
  import numpy
  import ray
  from contextlib import closing
  import logging
  import socket
  from zoo.orca.data.utils import ray_partitions_get_data_label
label_desc: Use image node_id 9 for calling the TFRunner obj's underlying member method code with example usage: obj.get_state() and returns: dict