Python Tensorflow扩展嵌套结构在训练期间断言失败

我正在尝试使用我自己的表格数据集基于Tensorflow的TFX创建我自己的TFX管道。因此，我只需用自己的数据集替换它们的数据集，并保留大部分现有代码

但是，当我到达培训组件-培训师时，由于嵌套结构不同，断言错误

assert\u same\u 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管道中发生了什么。

您使用的是什么类型的数据集。你能分享你的数据集让我们重现这个问题吗。谢谢您正在使用什么类型的数据集。你能分享你的数据集让我们重现这个问题吗。谢谢