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。

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

0
0

Google老师亲授 TensorFlow2.0 入门到进阶

Tensorflow2.0实战—以实战促理论的方式学习深度学习

1849 学习 · 896 问题

查看课程