获取'model.fit'keras API参数的值

Question

我正在尝试使用自定义回调函数获取 Kera 顺序模型的详细信息。我需要提取

model.fit()

API 中设置的参数的所有值，例如

batch_size

、

epochs

、

validation_split

等。但我无法在 Keras 的回调中访问它们。您知道如何自动获取这些值吗？ 我正在使用 Python 3.10 和 Keras 2.8。

Answer 1

经过一番研究，我发现我可以通过编写 Keras 回调来访问所有这些参数。事实上，我需要的所有参数都可以在

keras.callbacks

类中找到。这是一个示例脚本

class sample_calback(keras.callbacks.Callback):
    def __init__(self, model):
        super(sample_calback, self).__init__()

当我们在

model.fit

中使用这个回调时，可以从

self.params

字典中提取参数。希望这会对任何想做同样工作的人有所帮助。