Announcing the TensorFlow Dev Summit 2020 Learn more

tfx.components.BulkInferrer

View source on GitHub

Class BulkInferrer

A TFX component to do batch inference on a model with unlabelled examples.

Inherits From: BaseComponent

Aliases: tfx.components.bulk_inferrer.component.BulkInferrer

BulkInferrer consumes examples data and a model, and produces the inference results to an external location as PredictionLog proto.

BulkInferrer will infer on validated model.

Example

  # Uses BulkInferrer to inference on examples.
  bulk_inferrer = BulkInferrer(
      examples=example_gen.outputs['examples'],
      model=trainer.outputs['output'])

__init__

View source

__init__(
    examples=None,
    model=None,
    model_blessing=None,
    data_spec=None,
    model_spec=None,
    inference_result=None,
    instance_name=None
)

Construct an BulkInferrer component.

Args:

  • examples: A Channel of 'ExamplesPath' type, usually produced by ExampleGen component. required
  • model: A Channel of 'ModelExportPath' type, usually produced by Trainer component.
  • model_blessing: A Channel of 'ModelBlessingPath' type, usually produced by Model Validator component.
  • data_spec: bulk_inferrer_pb2.DataSpec instance that describes data selection.
  • model_spec: bulk_inferrer_pb2.ModelSpec instance that describes model specification.
  • inference_result: Channel of InferenceResult to store the inference results.
  • instance_name: Optional name assigned to this specific instance of BulkInferrer. Required only if multiple BulkInferrer components are declared in the same pipeline.

Child Classes

class DRIVER_CLASS

class SPEC_CLASS

Properties

component_id

DEPRECATED FUNCTION

component_type

DEPRECATED FUNCTION

downstream_nodes

exec_properties

id

Node id, unique across all TFX nodes in a pipeline.

If instance name is available, node_id will be: . otherwise, node_id will be:

Returns:

node id.

inputs

outputs

type

upstream_nodes

Methods

add_downstream_node

View source

add_downstream_node(downstream_node)

add_upstream_node

View source

add_upstream_node(upstream_node)

from_json_dict

View source

from_json_dict(
    cls,
    dict_data
)

Convert from dictionary data to an object.

to_json_dict

View source

to_json_dict()

Convert from an object to a JSON serializable dictionary.

Class Members

  • EXECUTOR_SPEC