tensorflow - 如何制作 tf.transform (Tensorflow Transform) 编码的字典？

Question

我正在尝试使用此tfx.components.Transform函数获取“tf.transform 编码字典” 。

 transform = Transform(
    examples=example_gen.outputs['examples'],
    schema=schema_gen.outputs['schema'],
    module_file=os.path.abspath(_taxi_transform_module_file),
    instance_name="taxi")
context.run(transform)

我需要这样的字典：“您加载的数据的字典（{feature_name：feature_value}）。”

如上所述的转换给了我一个 TfRecord 文件。我怎样才能正确解码？

任何帮助，将不胜感激。

Answer 1

import tensorflow_transform as tft
def preprocessing_fn(inputs):
   NUMERIC_FEATURE_KEYS = ['PetalLengthCm', 'PetalWidthCm', 
   'SepalLengthCm', 'SepalWidthCm']
   TARGET_FEATURES = "Species"
   outputs = inputs.copy()
   del outputs['Id']

   for key in NUMERIC_FEATURE_KEYS:
       outputs[key] = tft.scale_to_0_1(outputs[key])

 
   return outputs

写一个像这样的模块我已经为鸢尾花数据集写了一个它很容易理解你的数据集你也可以这样做它将被保存为一个 tfrecord 数据集