---
id: tensorflow-metadata
version: "1.21.0"
license: Apache 2.0
license_treatment: permissive
maintenance: active
---

# tensorflow-metadata — Library and standards for schema and statistics.

License: permissive · Maintenance: active · Downloads: 3.1M/mo

## What it is and what it does

TensorFlow Metadata is a library for defining and working with machine learning metadata in a standardized way. It provides protocol buffer-based representations for three key artifacts: schemas that describe the structure of tabular data (such as TensorFlow Examples), summary statistics computed over datasets, and problem statements that quantify model objectives. These representations can be created manually or generated automatically during data analysis workflows.

The library is designed to integrate with TensorFlow and TFX (TensorFlow Extended) pipelines, where metadata is consumed for data validation (checking that incoming data matches expected schemas), exploration (understanding dataset characteristics), and transformation (preparing data for model training). It sits at the intersection of data engineering and machine learning, providing a common language for describing data properties that both automated systems and human analysts need to understand.

Use it for:

- Define and validate schemas for tabular datasets before feeding them into TensorFlow training pipelines
- Generate and store summary statistics about training datasets for data drift detection and monitoring
- Document data requirements and constraints as part of a TFX workflow for reproducibility
- Automate data exploration by programmatically inspecting schema and statistics representations
- Share standardized metadata definitions across teams working on the same machine learning projects

## Worth the install?

AI-flagged interpretation of the facts above — verify before relying.

Provides standard schema, statistics, and problem statement representations for machine learning metadata that can be used with TensorFlow for data validation, exploration, and transformation.

Yes, if you are building machine learning pipelines with TensorFlow or TFX and need standardized metadata representations for schemas and statistics. The package is actively maintained, has no known vulnerabilities, low installation friction, and a permissive license. It is most valuable in team environments or complex pipelines where metadata standardization aids reproducibility and data governance; less critical for simple single-script experiments.

## Install

pip install tensorflow-metadata
uv add tensorflow-metadata
poetry add tensorflow-metadata

## Installing tensorflow-metadata

Before you install: Low friction installation with three lightweight runtime dependencies (absl-py, protobuf, googleapis-common-protos). Active maintenance with a recent release 66 days ago and ongoing repository activity.

License in practice: Apache 2.0 permissive license allows use in commercial and open-source projects with minimal restrictions; you must include a copy of the license and state significant changes.

Quickstart:

pip install tensorflow-metadata

import tensorflow_metadata as tfmd
from tensorflow_metadata.proto import schema_pb2

schema = schema_pb2.Schema()

Requires Python 3.10 or later (3.10, 3.11, 3.12, or 3.13)

Verify before relying:

- Whether the package works with TensorFlow versions other than those in its own dependency tree
- Whether summary statistics generation supports all common data types and distributions
- Whether schema validation performance scales to very large datasets

## Package facts

- License: Apache 2.0 (permissive)
- Python support: supports_current
- Install friction: low
- Maintenance: active
- Downloads: 3.1M/month (top 5,000 on PyPI)
- Known vulnerabilities: none known

## Tags

tensorflow metadata schema, ml data validation, dataset statistics representation, tfx metadata library, tabular data schema, machine learning data standards, tensorflow data exploration, tensorflow, data-validation, ml-metadata

[View on SkillFed](https://skillfed.io/packages/tensorflow-metadata) · [View on PyPI](https://pypi.org/project/tensorflow-metadata/)