終于搞懂了Keras中multiloss的對應關系介紹
更新時間:2020年06月22日 09:10:56 作者:evanna-y
這篇文章主要介紹了終于搞懂了Keras中multiloss的對應關系介紹,具有很好的參考價值,希望對大家有所幫助。一起跟隨小編過來看看吧
我就廢話不多說了,大家還是直接看代碼吧~
model = Model(inputs=[src, tgt], outputs=[y, flow]) #定義網(wǎng)絡的時候會給出輸入和輸出 model.compile(optimizer=Adam(lr=lr), loss=[ losses.cc3D(), losses.gradientLoss('l2')], loss_weights=[1.0, reg_param]) #訓練網(wǎng)絡的時候指定loss,如果是多l(xiāng)oss, loss weights分別對應前面的每個loss的權重,最后輸出loss的和 train_loss = model.train_on_batch( [X, atlas_vol], [atlas_vol, zero_flow]) #開始訓練,loss中y_pred 和y_true的對應關系是: #輸出y與atlas_vol算cc3Dloss,輸出flow與zero_flow算gradientloss
補充知識:keras服務器用fit_generator跑的代碼,loss,acc曲線圖的保存
我就廢話不多說了,大家還是直接看代碼吧~
import matplotlib.pyplot as plt ... //數(shù)據(jù)處理代碼 省略 history = model.fit_generator( image_generator, steps_per_epoch=2000 // 32 , epochs=16, verbose=1, validation_data=image_generator_TEST, validation_steps=20 ) print(history.history.keys()) plt.switch_backend('agg') #服務器上面保存圖片 需要設置這個 //acc plt.plot(history.history['acc']) plt.plot(history.history['val_acc']) plt.title('model accuracy') plt.ylabel('accuracy') plt.xlabel('epoch') plt.legend(['train', 'test'], loc='upper left') plt.savefig('acc.jpg') //loss plt.plot(history.history['loss']) plt.plot(history.history['val_loss']) plt.ylabel('loss') plt.xlabel('epoch') plt.legend(['train', 'test'], loc='upper left') plt.savefig('loss.jpg')
以上這篇終于搞懂了Keras中multiloss的對應關系介紹就是小編分享給大家的全部內(nèi)容了,希望能給大家一個參考,也希望大家多多支持腳本之家。
相關文章
Flask框架學習筆記(一)安裝篇(windows安裝與centos安裝)
Flask是一個輕量級的Web應用框架, 使用Python編寫。Flask也被稱為 “microframework” ,因為它使用簡單的核心,用 extension 增加其他功能。2014-06-06Python?pomegranate庫實現(xiàn)基于貝葉斯網(wǎng)絡拼寫檢查器
這篇文章主要為大家介紹了Python?pomegranate庫實現(xiàn)基于貝葉斯網(wǎng)絡拼寫檢查器示例詳解,有需要的朋友可以借鑒參考下,希望能夠有所幫助,祝大家多多進步,早日升職加薪<BR>2023-04-04keras 回調(diào)函數(shù)Callbacks 斷點ModelCheckpoint教程
這篇文章主要介紹了keras 回調(diào)函數(shù)Callbacks 斷點ModelCheckpoint教程,具有很好的參考價值,希望對大家有所幫助。一起跟隨小編過來看看吧2020-06-06SVM算法的理解及其Python實現(xiàn)多分類和二分類問題
這篇文章主要介紹了SVM算法的理解及其Python實現(xiàn)多分類和二分類問題,具有很好的參考價值,希望對大家有所幫助。如有錯誤或未考慮完全的地方,望不吝賜教2023-02-02django admin實現(xiàn)動態(tài)多選框表單的示例代碼
借助django-admin,可以快速得到CRUD界面,但若需要創(chuàng)建多選標簽字段時,需要對表單進行調(diào)整,本文通過示例代碼給大家介紹django admin多選框表單的實現(xiàn)方法,感興趣的朋友跟隨小編一起看看吧2021-05-05