Skip to main content

A library for maintaining metadata for artifacts.

Project description

ML Metadata

Python PyPI

ML Metadata (MLMD) is a library for recording and retrieving metadata associated with ML developer and data scientist workflows.

NOTE: ML Metadata may be backwards incompatible before version 1.0.

Getting Started

For more background on MLMD and instructions on using it, see the getting started guide

Installing from PyPI

The recommended way to install ML Metadata is to use the PyPI package:

pip install ml-metadata

Then import the relevant packages:

from ml_metadata import metadata_store
from ml_metadata.proto import metadata_store_pb2

Nightly Packages

ML Metadata (MLMD) also hosts nightly packages at https://pypi-nightly.tensorflow.org on Google Cloud. To install the latest nightly package, please use the following command:

pip install -i https://pypi-nightly.tensorflow.org/simple ml-metadata

Installing with Docker

This is the recommended way to build ML Metadata under Linux, and is continuously tested at Google.

Please first install docker and docker-compose by following the directions: docker; docker-compose.

Then, run the following at the project root:

DOCKER_SERVICE=manylinux-python${PY_VERSION}
sudo docker-compose build ${DOCKER_SERVICE}
sudo docker-compose run ${DOCKER_SERVICE}

where PY_VERSION is one of {36, 37, 38}.

A wheel will be produced under dist/, and installed as follows:

pip install dist/*.whl

Installing from source

1. Prerequisites

To compile and use ML Metadata, you need to set up some prerequisites.

Install Bazel

If Bazel is not installed on your system, install it now by following these directions.

Install cmake

If cmake is not installed on your system, install it now by following these directions.

2. Clone ML Metadata repository

git clone https://github.com/google/ml-metadata
cd ml-metadata

Note that these instructions will install the latest master branch of ML Metadata. If you want to install a specific branch (such as a release branch), pass -b <branchname> to the git clone command.

3. Build the pip package

ML Metadata uses Bazel to build the pip package from source:

python setup.py bdist_wheel

You can find the generated .whl file in the dist subdirectory.

4. Install the pip package

pip install dist/*.whl

5.(Optional) Build the grpc server

ML Metadata uses Bazel to build the c++ binary from source:

bazel build -c opt --define grpc_no_ares=true  //ml_metadata/metadata_store:metadata_store_server

Supported platforms

MLMD is built and tested on the following 64-bit operating systems:

  • macOS 10.14.6 (Mojave) or later.
  • Ubuntu 16.04 or later.
  • Windows 7 or later.

Download files

Download the file for your platform. If you're not sure which to choose, learn more about installing packages.

Files for ml-metadata, version 1.0.0
Filename, size File type Python version Upload date Hashes
Filename, size ml_metadata-1.0.0-cp36-cp36m-macosx_10_9_x86_64.whl (5.1 MB) File type Wheel Python version cp36 Upload date Hashes View
Filename, size ml_metadata-1.0.0-cp36-cp36m-manylinux_2_12_x86_64.manylinux2010_x86_64.whl (2.9 MB) File type Wheel Python version cp36 Upload date Hashes View
Filename, size ml_metadata-1.0.0-cp36-cp36m-win_amd64.whl (1.5 MB) File type Wheel Python version cp36 Upload date Hashes View
Filename, size ml_metadata-1.0.0-cp37-cp37m-macosx_10_9_x86_64.whl (5.1 MB) File type Wheel Python version cp37 Upload date Hashes View
Filename, size ml_metadata-1.0.0-cp37-cp37m-manylinux_2_12_x86_64.manylinux2010_x86_64.whl (2.9 MB) File type Wheel Python version cp37 Upload date Hashes View
Filename, size ml_metadata-1.0.0-cp37-cp37m-win_amd64.whl (1.5 MB) File type Wheel Python version cp37 Upload date Hashes View
Filename, size ml_metadata-1.0.0-cp38-cp38-macosx_10_9_x86_64.whl (5.1 MB) File type Wheel Python version cp38 Upload date Hashes View
Filename, size ml_metadata-1.0.0-cp38-cp38-manylinux_2_12_x86_64.manylinux2010_x86_64.whl (2.9 MB) File type Wheel Python version cp38 Upload date Hashes View
Filename, size ml_metadata-1.0.0-cp38-cp38-win_amd64.whl (1.5 MB) File type Wheel Python version cp38 Upload date Hashes View

Supported by

AWS AWS Cloud computing Datadog Datadog Monitoring DigiCert DigiCert EV certificate Facebook / Instagram Facebook / Instagram PSF Sponsor Fastly Fastly CDN Google Google Object Storage and Download Analytics Microsoft Microsoft PSF Sponsor Pingdom Pingdom Monitoring Salesforce Salesforce PSF Sponsor Sentry Sentry Error logging StatusPage StatusPage Status page