Мультиклассификация NN с ошибкой Кераса

Question

Я получаю ошибку при попытке сделать мультиклассификацию с тремя классами.

Error: TypeError: fit_generator() got multiple values for argument 'steps_per_epoch'

Ошибка выдачи кода:

NN.fit_generator(
                        train_set, train_labels,
                        steps_per_epoch=(train_samples/ batch_size),
                        epochs=epochs,
                        validation_data=(validation_set, validation_labels),
                        validation_steps=(validation_samples / batch_size))

Полный код: https://pastebin.com/V1YwJW3X

Я был бы очень признателен за любую помощь в этом вопросе, так как я в полной растерянности. Спасибо!

Answer 1

Документация Keras содержит следующее определение для fit_generator:

fit_generator(self, generator, steps_per_epoch=None, epochs=1, verbose=1, callbacks=None, validation_data=None, validation_steps=None, class_weight=None, max_queue_size=10, workers=1, use_multiprocessing=False, shuffle=True, initial_epoch=0)

Вы предоставили два позиционных аргумента:

• train_set - это присвоено generator
• train_labels - это присвоено steps_per_epoch

Но затем вы предоставляете другой (теперь ключевой аргумент) steps_per_epoch, следовательно, ошибка.