Keras中的TypeError：即使已经提供了shuffle =“batch”，也要传递shuffle =“batch”

Posted 2021-05-04

我正在尝试使用tf.keras的fit（）方法训练我的模型，因为输入数据来自hdf5文件，我将参数shuffle ='batch'传递给fit（）方法。但在第一个纪元结束后，出现以下错误：

TypeError: TypeError while preparing batch. If using HDF5 input data, pass shuffle="batch".

这是我的fit（）方法：

model.fit(
    x=features_train,
    y=topics_train,
    batch_size=16384,
    epochs=35,
    callbacks=create_callbacks(),
    validation_data=(features_val, topics_val),
    shuffle='batch'
)

变量features_train和features_val取自hdf5文件。

答案

通过将features_val转换为numpy数组解决了这个问题。

features_val_arr = np.array(features_val)

model.fit(
    x=features_train,
    y=topics_train,
    batch_size=16384,
    epochs=35,
    callbacks=create_callbacks(),
    validation_data=(features_val_arr, topics_val),
    shuffle='batch'
)