亚洲在线久爱草,狠狠天天香蕉网,天天搞日日干久草,伊人亚洲日本欧美

為了賬號安全,請及時綁定郵箱和手機立即綁定
已解決430363個問題,去搜搜看,總會有你想問的

將 keras 模型作為參數發送給 Python 中的外部函數

將 keras 模型作為參數發送給 Python 中的外部函數

慕碼人8056858 2021-12-16 16:09:37
我想在訓練后將 Keras 模型發送到保存在另一個 python 文件中的另一個 python 函數?如何將模型作為參數發送?謝謝你。
查看完整描述

1 回答

?
子衿沉夜

TA貢獻1828條經驗 獲得超3個贊

如果我理解正確,您想將在腳本 A 中創建的模型傳輸到腳本 B 以便它可以在那里使用。


根據我的經驗,在不同的腳本中使用 Keras 模型的最簡單方法是將模型作為文件保存到磁盤。這里描述的Keras文檔:。


from keras.models import load_model


model.save('my_model.h5')  # creates a HDF5 file 'my_model.h5'

del model  # deletes the existing model


# returns a compiled model

# identical to the previous one

model = load_model('my_model.h5')

將模型傳遞到不同的 Python 文件(即通過命令行參數),然后可以通過將保存該模型的文件名傳遞給第二個腳本來完成。然后,此腳本可以從磁盤加載模型并使用它。


如果您一次只有 1 個模型,您可以選擇一個文件名并將其硬編碼到您的函數中。例如:


腳本A


# assuming you already have a model stored in 'model'

model.save('my_stored_model.h5')

腳本 B(訪問保存的模型)


from keras.models import load_model


def function_a():

    model = load_model('my_stored_model.h5')

    return model.predict(...)


查看完整回答
反對 回復 2021-12-16
  • 1 回答
  • 0 關注
  • 181 瀏覽
慕課專欄
更多

添加回答

舉報

0/150
提交
取消
微信客服

購課補貼
聯系客服咨詢優惠詳情

幫助反饋 APP下載

慕課網APP
您的移動學習伙伴

公眾號

掃描二維碼
關注慕課網微信公眾號