fit函数是怎么找到数据集里面的特征和标签的?
来源:4-8 tf.data读取tfrecord文件并与tf.keras结合使用
刘鹏4
2021-08-03
fit函数是如何找到数据集里面的特征和标签的?是不是我的特征feature必须的命名为“input_features”,标签必须要命名为“label”
写回答
1回答
-
正十七
2021-08-07
不用这样命名,参考函数定义:https://www.tensorflow.org/api_docs/python/tf/keras/Model#fit
如果传进去的是直接的数据,那么就是x,y, x是输入,y是label。
如果传进去的是tf.data.Dataset, 那么dataset里的每组数据都是<x, y> 或者 <x, y, weights>,其中x是输入,y是label。
特征明明只是为了我们开发者的可读性。
00
相似问题