class_docstr: Callback for Keras models.
code: def on_train_end(self, logs=None):
      self.train_finish_time = time.time()
  
imports: from __future__ import absolute_import
  from __future__ import division
  from __future__ import print_function
  import time
  import tensorflow
  from tensorflow.core.protobuf import rewriter_config_pb2
  from tensorflow.python import tf2
  from tensorflow.python.eager import profiler
label_desc: Use image node_id 2 for calling the TimeHistory obj's underlying member method code with example usage: obj.on_train_end(logs) without return types