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(...)
添加回答
舉報