具有 Keras 错误的多分类 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 documentation 为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，因此出现了错误。