使用tensorflow数据集和keras进行验证

时间:2018-08-21 13:52:38

标签: python-3.x tensorflow keras

我有两个数据集

dataset_train = tf.data.Dataset.from_tensor_slices(((x_train, mask_train), y_train))
dataset_dev   = tf.data.Dataset.from_tensor_slices(((x_dev, mask_dev), y_dev))

和一个keras模型

model = tf.keras.Model(...)

我想将model上的dataset_train放入并在dataset_dev上进行验证。

model.compile('adam', loss='mse', metrics=['accuracy'])

model.fit(x=dataset_train, 
          epochs=args.epoch_num, 
          steps_per_epoch=1, 
          validation_data=dataset_dev,
          validation_steps=1)

我收到以下错误:

ValueError: ('Error when checking model input: expected no data, but got:', (<tf.Tensor 'IteratorGetNext_1:0' shape=(?, 512, 64, 12) dtype=float64>, <tf.Tensor 'IteratorGetNext_1:1' shape=(?, 512, 64, 1) dtype=float64>))

但是当我删除验证数据时,一切正常

model.fit(x=dataset_train, 
          epochs=args.epoch_num, 
          steps_per_epoch=1)

知道我在做什么错吗?

0 个答案:

没有答案