如果您保存了具有完整架构及其训练状态的模型。即你用过这样的东西。
model.save('myfirstmodel.h5')
你可以使用
pprint(model.to_json())
pprint(model.to_yaml())
json 的输出:
('{"class_name": "Sequential", "config": {"name": "sequential", "layers": '
'[{"class_name": "InputLayer", "config": {"batch_input_shape": [null, 13], '
'"dtype": "float32", "sparse": false, "ragged": false, "name": "d1_input"}}, '
'{"class_name": "Dense", "config": {"name": "d1", "trainable": true, '
'"batch_input_shape": [null, 13], "dtype": "float32", "units": 4, '
'"activation": "relu", "use_bias": true, "kernel_initializer": {"class_name": '
'"Ones", "config": {}}, "bias_initializer": {"class_name": "Zeros", "config": '
'{}}, "kernel_regularizer": null, "bias_regularizer": null, '
'"activity_regularizer": null, "kernel_constraint": null, "bias_constraint": '
'null}}, {"class_name": "Dense", "config": {"name": "d2", "trainable": true, '
'"dtype": "float32", "units": 6, "activation": "relu", "use_bias": true, '
'"kernel_initializer": {"class_name": "GlorotUniform", "config": {"seed": '
'null}}, "bias_initializer": {"class_name": "Zeros", "config": {}}, '
'"kernel_regularizer": null, "bias_regularizer": null, '
'"activity_regularizer": null, "kernel_constraint": null, "bias_constraint": '
'null}}, {"class_name": "Dropout", "config": {"name": "dropout", "trainable": '
'true, "dtype": "float32", "rate": 0.2, "noise_shape": null, "seed": null}}, '
'{"class_name": "Dense", "config": {"name": "out", "trainable": true, '
'"dtype": "float32", "units": 2, "activation": "sigmoid", "use_bias": true, '
'"kernel_initializer": {"class_name": "GlorotUniform", "config": {"seed": '
'null}}, "bias_initializer": {"class_name": "Zeros", "config": {}}, '
'"kernel_regularizer": null, "bias_regularizer": null, '
'"activity_regularizer": null, "kernel_constraint": null, "bias_constraint": '
'null}}]}, "keras_version": "2.4.0", "backend": "tensorflow"}')
但是,如果您有一个冻结模型,其中您的正常方法不起作用,您可以使用 netron 查看模型的结构。
它显示了分层架构以及使用了哪些激活函数、参数及其权重。您可以将这些权重下载为 NumPy 数组。
您可以使用 Netron 来查找模型的架构以及权重。使用这些结构信息,您可以重建模型。
见Link。
你会得到这样的输出: