A TFX SchemaGen component to generate a schema from the training data.
tfx.v1.components.SchemaGen(
statistics: tfx.v1.types.BaseChannel
,
infer_feature_shape: Optional[Union[bool, tfx.v1.dsl.experimental.RuntimeParameter
]] = True,
exclude_splits: Optional[List[str]] = None
)
Used in the notebooks
Used in the tutorials |
---|
The SchemaGen component uses TensorFlow Data Validation to generate a schema from input statistics. The following TFX libraries use the schema:
- TensorFlow Data Validation
- TensorFlow Transform
- TensorFlow Model Analysis
In a typical TFX pipeline, the SchemaGen component generates a schema which is consumed by the other pipeline components.
Example
# Generates schema based on statistics files.
infer_schema = SchemaGen(statistics=statistics_gen.outputs['statistics'])
Component outputs
contains:
schema
: Channel of typestandard_artifacts.Schema
for schema result.
See the SchemaGen guide for more details.
Attributes | |
---|---|
outputs
|
Component's output channel dict. |