/sample-tfx-pipeline

A sample TFX pipeline intended to run in local

Primary LanguagePythonApache License 2.0Apache-2.0

ML Ops with TFX

Setup

For this project, you will need Python 3.7, 3.8 or 3.9, with a virtualenv with the dependencies included in the file requirements.txt.

Find instructions below to install the dependencies:

Please don't use Python < 3.7 (e.g. 3.6) or Python > 3.9 (e.g. 3.10), they will not work with TFX. For more details, please check:

At the moment of writing this, the Cloud Shell has Python 3.9. You can check your Python version by running the following command:

python --version

Once you have made sure you have the correct Python version, create a virtualenv:

python -m venv tfxenv

Activate it:

source ./tfxenv/bin/activate

And install the dependencies in the file requirements.txt, by running:

pip install -r requirements.txt

Running the pipeline

Edit the scripts in the directory scripts just in case you want to adapt any of the default options.

The playground branch of this repository contains incomplete code that you need to finish, as an exercise to learn the ropes of TFX pipelines.

To run the pipeline, run this script:

./scripts/launch_local.sh