簡體   English   中英

Keras Functional API更改每個API中的圖層名稱

[英]Keras Functional API changing layer names in every API

當我在模型中運行功能性API進行k折交叉驗證時,在每次折疊的返回擬合模型中,命名密集層的數字都會增加。 就像在第一折中是dense_2_acc ,在第二折中是dense_5_acc

通過我的模型摘要顯示我的模型是正確的。 為什么要更改每個折疊的擬合模型history對象中的名稱?

在此處輸入圖片說明

這是一個非常好的問題,它說明了有關keras一些真正重要的問題。 名稱以這種方式更改的原因是,即使您覆蓋模型, keras也不會清除先前定義的變量。 您可以通過調用以下命令輕松檢查變量是否仍在session.graph

from keras import backend as K
K.get_session().graph.get_collection('variables')

為了清除先前的模型變量,可以調用:

K.clear_session()

但是-要小心-因為您可能會丟失現有模型。 如果要保持名稱不變,只需在圖層實例中添加name參數即可為圖層name ,例如:

Dense(10, activation='softmax', name='output')

暫無
暫無

聲明:本站的技術帖子網頁,遵循CC BY-SA 4.0協議,如果您需要轉載,請注明本站網址或者原文地址。任何問題請咨詢:yoyou2525@163.com.

 
粵ICP備18138465號  © 2020-2024 STACKOOM.COM