首页 > 解决方案 > 如何使用 tf.data.experimental.make_csv_dataset 生成(输入、目标、样本权重)元组

问题描述

我正在使用 make_csv_dataset 函数为模型训练提供数据。当我使用 tf.keras.Model.fit 函数时,我需要从 tf.data 对象为 fit 函数提供 (inputs,targets,sample_weights) 元组。如何使用 make_csv_dataset 函数来生成这个元组?(我的 csv 在其他特征列旁边有一个权重列和一个标签列)

目前 make_csv_dataset 中只有参数 label_name 来指示哪个是标签列,但是如何指示此 tf.data 对象的权重列?

对于这个问题,任何解决方法都是可以接受的,在此先感谢!

标签: pythontensorflow

解决方案


推荐阅读