Python tf.keras.optimizers.Optimizer.get_weights用法及代碼示例

用法

get_weights()

返回優化器的當前權重。

優化器的權重是它的狀態(即變量)。此函數將與此優化器關聯的權重值作為 Numpy 數組列表返回。第一個值始終是優化器的迭代計數，然後是優化器的狀態變量，按照它們的創建順序排列。返回的列表又可以用於將狀態加載到類似的參數化優化器中。

例如，這個簡單模型的 RMSprop 優化器返回一個包含三個值的列表——迭代計數，然後是內核的均方根值和單個 Dense 層的偏差：

opt = tf.keras.optimizers.RMSprop()
m = tf.keras.models.Sequential([tf.keras.layers.Dense(10)])
m.compile(opt, loss='mse')
data = np.arange(100).reshape(5, 20)
labels = np.zeros(5)
print('Training'); results = m.fit(data, labels)
Training ...
len(opt.get_weights())
3