具有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，因此错误。