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

问题描述 投票:1回答:1

我正在尝试使用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_trainfeatures_val取自hdf5文件。

keras deep-learning hdf5 tf.keras
1个回答
1
投票

通过将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'
)
© www.soinside.com 2019 - 2024. All rights reserved.