Missed TensorFlow Dev Summit? Check out the video playlist. Watch recordings

Module: tfx.components.trainer.executor

View source on GitHub

TFX local trainer executor.


class Executor: Local estimator based trainer executor used by the TFX Trainer component.

class GenericExecutor: Local generic trainer executor for the TFX Trainer component.

class TrainerFnArgs: Wrapper class to help migrate from contrib.HParam to new data structure.

Other Members

  • BASE_MODEL_KEY = 'base_model'
  • EXAMPLES_KEY = 'examples'
  • HYPERPARAMETERS_KEY = 'hyperparameters'
  • OUTPUT_MODEL_KEY = 'model'
  • SCHEMA_KEY = 'schema'
  • TRANSFORM_GRAPH_KEY = 'transform_graph'