python - Tensorflow Extended - 嵌套结构在训练期间断言失败

Question

我正在尝试使用我自己的表格数据集基于 Tensorflow 的 TFX教程创建自己的 TFX 管道。所以我只是用我自己的数据集替换他们的数据集，并保留大部分现有代码。

然而，当我到达训练组件 - Trainer -assert_same_structure由于不同的嵌套结构，断言错误失败：

ValueError: The two structures don't have the same nested structure.
First structure: type=SparseTensorSpec str=SparseTensorSpec(TensorShape([40, None]), tf.int64)
Second structure: type=Tensor str=Tensor("Address:0", shape=(None,), dtype=int64)
More specifically: Substructure "type=SparseTensorSpec str=SparseTensorSpec(TensorShape([40, None]), tf.int64)" is a sequence, while substructure "type=Tensor str=Tensor("Address:0", shape=(None,), dtype=int64)" is not
Entire first structure:
.
Entire second structure:
.

有没有人遇到过这个问题？很难弄清楚 TFX 管道后台发生了什么。