Keras和make_csv_dataset的兼容性

时间:2018-08-02 07:24:35

标签: tensorflow keras

tf.contrib.data.make_csv_dataset()可以在Tensorflow 1.9.0中用于Keras模型吗?

1 个答案:

答案 0 :(得分:2)

是的,tf.contrib.data.make_csv_dataset()返回一个tf.data.Dataset,您可以将tf.data.Dataset传递给Keras模型的fit方法。 在此处查看一些示例: https://www.tensorflow.org/guide/keras#input_tfdata_datasets