'TFX - What is example_gen_pb2 and where is it documented?

The ExampleGen TFX Pipeline Component has the example code which uses example_gen_pb2 but what is it and where is it documented?

# Input has a single split 'input_dir/*'.
# Output 2 splits: train:eval=3:1.
output = proto.Output(
             split_config=example_gen_pb2.SplitConfig(splits=[
                 proto.SplitConfig.Split(name='train', hash_buckets=3),
                 proto.SplitConfig.Split(name='eval', hash_buckets=1)
             ]))
example_gen = CsvExampleGen(input_base=input_dir, output_config=output)
tfx


Solution 1:[1]

If you want to import it, you can do it from tfx.proto:

from tfx.proto import example_gen_pb2

Source: https://github.com/tensorflow/tfx/blob/master/tfx/components/example_gen/utils_test.py

Sources

This article follows the attribution requirements of Stack Overflow and is licensed under CC BY-SA 3.0.

Source: Stack Overflow

Solution Source
Solution 1 Lahkim Omar

Related Questions