如何获取tensorflow keras模型的所有细节

Question

我想知道如何查看与keras模型相关的所有细节。显然，除了输出形状和可优化参数的数量之外，model.summary注释并没有给出所有的细节。如何获取有关正则化值和丢弃层等的信息。

Answer 1

您可以简单地使用model.summary()来打印网络参数的数量以及层。您可以使用keras.utils.plot_model(model_name, "file_name")绘制模型的图形。

要检查您的模型，您可以使用像.add_graph()方法这样的tensorboard实用程序。