求助：基于TensorFlow Layers API的非Estimator鸢尾花分类器无法运行

最近我在学习TensorFlow，为了练手，打算训练一个鸢尾花数据集的分类器。参考TensorFlow官方的鸢尾花示例网络架构，我尝试用Layers API来复现这个模型：

• 包含3个隐藏层，神经元数量分别为10、20、10
• 因为鸢尾花是三分类任务，最后设置了一个输出层：神经元数量为3，激活函数采用softmax的全连接层

对应的代码片段如下：

def parse_csv(line):
    data = tf.decode_csv(...)
    # 后续处理逻辑省略

内容的提问来源于stack exchange，提问作者Gianluca Micchi