Autoencoder, steps_per_epoch, какое значение он должен содержать?
Я пытаюсь создать составной автоэнкодер с моим собственным набором данных, все отлично работает, когда я пытаюсь нарисовать кривую с помощью тензорной доски, я получаю следующие скаляры:
я думаю, что ошибка находится в steps_per_epoch, если это не X_train.shape[0], так что он должен содержать:
autoencoder.fit_generator(generated_data.flow(X_train, X_train, batch_size=batch_size), steps_per_epoch=X_train.shape[0], epochs=epochs, validation_data=(X_test, X_test), callbacks=[TensorBoard(log_dir='/tmp/autoencoder')])
и другое, как я могу добавить Accuarcy?
@gorjan спасибо за ответ, я отредактировал свой код
Ответы 1
Из документации fit_generator
steps_per_epoch: Integer. Total number of steps (batches of samples) to yield from generator before declaring one epoch finished and starting the next epoch. It should typically be equal to the number of samples of your dataset divided by the batch size. Optional for Sequence: if unspecified, will use the len(generator) as a number of steps.
Таким образом, вы должны установить его примерно равным X_train.shape[0]/batch_size.
Для контроля точности используйте
autoencoder.compile(optimizer='rmsprop', loss='mse', metrics=['mse', 'accuracy'])
Большое спасибо, это то, что я искал.
для графика Проблема была в callbacks=[TensorBoard(log_dir='/tmp/autoencoder')] тоже у меня есть два кода, которые вызывают один и тот же log_dir
Можете ли вы опубликовать код, где вы пишете резюме?