fit函数是怎么找到数据集里面的特征和标签的？

Question

fit函数是如何找到数据集里面的特征和标签的？是不是我的特征feature必须的命名为“input_features”,标签必须要命名为“label”

Answer 1

正十七 2021-08-07 16:09:58

不用这样命名，参考函数定义：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。

特征明明只是为了我们开发者的可读性。