TFX - 什么是 example_gen_pb2 以及它記錄在何處？

Question

ExampleGen TFX 管道組件具有使用example_gen_pb2的示例代碼，但它是什么以及記錄在哪里？

# 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)

Answer 1

API 文檔在這里。 由於它是原型文件，最簡單的方法是檢查定義本身，其中包含對數據結構的注釋。
謝謝！

Answer 2

如果你想導入它，你可以從 tfx.proto 中導入：

from tfx.proto import example_gen_pb2

來源： https://github.com/tensorflow/tfx/blob/master/tfx/components/example_gen/utils_test.py