TypeError в Keras: pass shuffle = "batch", даже если shuffle = "batch" уже предоставлен - PullRequest
1 голос
/ 14 апреля 2019

Я пытаюсь обучить мою модель, используя метод fit () с tf.keras, поскольку входные данные взяты из файла 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.

1 Ответ

1 голос
/ 14 апреля 2019

Решил это путем преобразования 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'
)
...