##### Copyright © 2019 Google Inc.

*The Feature Engineering Component of TensorFlow Extended (TFX)*

This example colab notebook provides a very simple example of how TensorFlow Transform (`tf.Transform`

) can be used to preprocess data using exactly the same code for both training a model and serving inferences in production.

TensorFlow Transform is a library for preprocessing input data for TensorFlow, including creating features that require a full pass over the training dataset. For example, using TensorFlow Transform you could:

- Normalize an input value by using the mean and standard deviation
- Convert strings to integers by generating a vocabulary over all of the input values
- Convert floats to integers by assigning them to buckets, based on the observed data distribution

TensorFlow has built-in support for manipulations on a single example or a batch of examples. `tf.Transform`

extends these capabilities to support full passes over the entire training dataset.

The output of `tf.Transform`

is exported as a TensorFlow graph which you can use for both training and serving. Using the same graph for both training and serving can prevent skew, since the same transformations are applied in both stages.

## Python check and imports

First, we'll make sure that we're using Python 2. Then, we'll go ahead and install and import the stuff we need.

```
from __future__ import print_function
import sys, os
# Confirm that we're using Python 2
assert sys.version_info.major is 2, 'Oops, not running Python 2'
```

```
try:
import tensorflow_transform as tft
except ImportError:
# this will take a minute, ignore the warnings
!pip install -q tensorflow-transform
from __future__ import absolute_import
from __future__ import division
from __future__ import print_function
import pprint
import tempfile
import tensorflow as tf
import tensorflow_transform as tft
import tensorflow_transform.beam.impl as tft_beam
from tensorflow_transform.tf_metadata import dataset_metadata
from tensorflow_transform.tf_metadata import dataset_schema
```

DEPRECATION: Python 2.7 will reach the end of its life on January 1st, 2020. Please upgrade your Python as Python 2.7 won't be maintained after that date. A future version of pip will drop support for Python 2.7.

## Data: Create some dummy data

We'll create some simple dummy data for our simple example:

`raw_data`

is the initial raw data that we're going to preprocess`raw_data_metadata`

contains the schema that tells us the types of each of the columns in`raw_data`

. In this case, it's very simple.

```
raw_data = [
{'x': 1, 'y': 1, 's': 'hello'},
{'x': 2, 'y': 2, 's': 'world'},
{'x': 3, 'y': 3, 's': 'hello'}
]
raw_data_metadata = dataset_metadata.DatasetMetadata(
dataset_schema.from_feature_spec({
'y': tf.FixedLenFeature([], tf.float32),
'x': tf.FixedLenFeature([], tf.float32),
's': tf.FixedLenFeature([], tf.string),
}))
```

## Transform: Create a preprocessing function

The *preprocessing function* is the most important concept of tf.Transform. A preprocessing function is where the transformation of the dataset really happens. It accepts and returns a dictionary of tensors, where a tensor means a `Tensor`

or `SparseTensor`

. There are two main groups of API calls that typically form the heart of a preprocessing function:

**TensorFlow Ops:**Any function that accepts and returns tensors, which usually means TensorFlow ops. These add TensorFlow operations to the graph that transforms raw data into transformed data one feature vector at a time. These will run for every example, during both training and serving.**TensorFlow Transform Analyzers:**Any of the analyzers provided by tf.Transform. Analyzers also accept and return tensors, but unlike TensorFlow ops they only run once, during training, and typically make a full pass over the entire training dataset. They create tensor constants, which are added to your graph. For example,`tft.min`

computes the minimum of a tensor over the training dataset. tf.Transform provides a fixed set of analyzers, but this will be extended in future versions.

```
def preprocessing_fn(inputs):
"""Preprocess input columns into transformed columns."""
x = inputs['x']
y = inputs['y']
s = inputs['s']
x_centered = x - tft.mean(x)
y_normalized = tft.scale_to_0_1(y)
s_integerized = tft.compute_and_apply_vocabulary(s)
x_centered_times_y_normalized = (x_centered * y_normalized)
return {
'x_centered': x_centered,
'y_normalized': y_normalized,
's_integerized': s_integerized,
'x_centered_times_y_normalized': x_centered_times_y_normalized,
}
```

## Putting it all together

Now we're ready to transform our data. We'll use Apache Beam with a direct runner, and supply three inputs:
1. `raw_data`

- The raw input data that we created above
2. `raw_data_metadata`

- The schema for the raw data
3. `preprocessing_fn`

- The function that we created to do our transformation

```
def main():
# Ignore the warnings
with tft_beam.Context(temp_dir=tempfile.mkdtemp()):
transformed_dataset, transform_fn = ( # pylint: disable=unused-variable
(raw_data, raw_data_metadata) | tft_beam.AnalyzeAndTransformDataset(
preprocessing_fn))
transformed_data, transformed_metadata = transformed_dataset # pylint: disable=unused-variable
print('\nRaw data:\n{}\n'.format(pprint.pformat(raw_data)))
print('Transformed data:\n{}'.format(pprint.pformat(transformed_data)))
if __name__ == '__main__':
main()
```

INFO:tensorflow:Assets added to graph. INFO:tensorflow:No assets to write. INFO:tensorflow:SavedModel written to: /tmp/tmpxUkIXS/tftransform_tmp/7fe0892ee66247829fd2d3fc48ea7da6/saved_model.pb INFO:tensorflow:Assets added to graph. INFO:tensorflow:No assets to write. INFO:tensorflow:SavedModel written to: /tmp/tmpxUkIXS/tftransform_tmp/57d387adedc846098be65ee85bd9a969/saved_model.pb INFO:tensorflow:Saver not created because there are no variables in the graph to restore INFO:tensorflow:Saver not created because there are no variables in the graph to restore INFO:tensorflow:Saver not created because there are no variables in the graph to restore INFO:tensorflow:Saver not created because there are no variables in the graph to restore INFO:tensorflow:Assets added to graph. INFO:tensorflow:Assets added to graph. INFO:tensorflow:Assets written to: /tmp/tmpxUkIXS/tftransform_tmp/cfdb90c600814fa9b94746f13bdef09f/assets INFO:tensorflow:Assets written to: /tmp/tmpxUkIXS/tftransform_tmp/cfdb90c600814fa9b94746f13bdef09f/assets INFO:tensorflow:SavedModel written to: /tmp/tmpxUkIXS/tftransform_tmp/cfdb90c600814fa9b94746f13bdef09f/saved_model.pb INFO:tensorflow:SavedModel written to: /tmp/tmpxUkIXS/tftransform_tmp/cfdb90c600814fa9b94746f13bdef09f/saved_model.pb WARNING:tensorflow:Expected binary or unicode string, got type_url: "type.googleapis.com/tensorflow.AssetFileDef" value: "\n\013\n\tConst_2:0\022-vocab_compute_and_apply_vocabulary_vocabulary" WARNING:tensorflow:Expected binary or unicode string, got type_url: "type.googleapis.com/tensorflow.AssetFileDef" value: "\n\013\n\tConst_2:0\022-vocab_compute_and_apply_vocabulary_vocabulary" INFO:tensorflow:Saver not created because there are no variables in the graph to restore INFO:tensorflow:Saver not created because there are no variables in the graph to restore WARNING:tensorflow:Expected binary or unicode string, got type_url: "type.googleapis.com/tensorflow.AssetFileDef" value: "\n\013\n\tConst_2:0\022-vocab_compute_and_apply_vocabulary_vocabulary" WARNING:tensorflow:Expected binary or unicode string, got type_url: "type.googleapis.com/tensorflow.AssetFileDef" value: "\n\013\n\tConst_2:0\022-vocab_compute_and_apply_vocabulary_vocabulary" INFO:tensorflow:Saver not created because there are no variables in the graph to restore INFO:tensorflow:Saver not created because there are no variables in the graph to restore Raw data: [{'s': 'hello', 'x': 1, 'y': 1}, {'s': 'world', 'x': 2, 'y': 2}, {'s': 'hello', 'x': 3, 'y': 3}] Transformed data: [{u's_integerized': 0, u'x_centered': -1.0, u'x_centered_times_y_normalized': -0.0, u'y_normalized': 0.0}, {u's_integerized': 1, u'x_centered': 0.0, u'x_centered_times_y_normalized': 0.0, u'y_normalized': 0.5}, {u's_integerized': 0, u'x_centered': 1.0, u'x_centered_times_y_normalized': 1.0, u'y_normalized': 1.0}]

## Is this the right answer?

Previously, we used `tf.Transform`

to do this:

```
x_centered = x - tft.mean(x)
y_normalized = tft.scale_to_0_1(y)
s_integerized = tft.compute_and_apply_vocabulary(s)
x_centered_times_y_normalized = (x_centered * y_normalized)
```

#### x_centered

With input of `[1, 2, 3]`

the mean of x is 2, and we subtract it from x to center our x values at 0. So our result of `[-1.0, 0.0, 1.0]`

is correct.

#### y_normalized

We wanted to scale our y values between 0 and 1. Our input was `[1, 2, 3]`

so our result of `[0.0, 0.5, 1.0]`

is correct.

#### s_integerized

We wanted to map our strings to indexes in a vocabulary, and there were only 2 words in our vocabulary ("hello" and "world"). So with input of `["hello", "world", "hello"]`

our result of `[0, 1, 0]`

is correct.

#### x_centered_times_y_normalized

We wanted to create a new feature by crossing `x_centered`

and `y_normalized`

using multiplication. Note that this multiplies the results, not the original values, and our new result of `[-0.0, 0.0, 1.0]`

is correct.