How do I get the weights of a layer in Keras?

前端 未结 4 375
误落风尘
误落风尘 2020-12-08 02:38

I am using Windows 10, Python 3.5, and tensorflow 1.1.0. I have the following script:

import tensorflow as tf
import tensorflow.contrib.keras.api.keras.backe         


        
4条回答
  •  Happy的楠姐
    2020-12-08 03:16

    If you want to get weights and biases of all layers, you can simply use:

    for layer in model.layers: print(layer.get_config(), layer.get_weights())
    

    This will print all information that's relevant.

    If you want the weights directly returned as numpy arrays, you can use:

    first_layer_weights = model.layers[0].get_weights()[0]
    first_layer_biases  = model.layers[0].get_weights()[1]
    second_layer_weights = model.layers[1].get_weights()[0]
    second_layer_biases  = model.layers[1].get_weights()[1]
    

    etc.

提交回复
热议问题