iT邦幫忙

第 12 屆 iThome 鐵人賽

DAY 24
2
影片教學

全民瘋AI系列系列 第 24

[Day 24] 儲存訓練好的模型

儲存訓練好的模型

今日學習目標

  • 使用 pickle + gzip 儲存模型
    • 將訓練好的模型打包並儲存
  • 載入儲存的模型
    • 讀取打包好的模型,並預測

我們訓練好了一個模型以後需要保存並讓下一次直接預測。常見的兩種保存Model的模塊有 pickle 與 joblib,範例使用 pickle 。由於 pickle 儲存模型後容量可能會有好幾百MB因此建議可以透過 gzip 來壓縮模型並儲存。

本系列教學簡報 PDF & Code 都可以從我的 GitHub 取得!


上一篇
[Day 23] 免費雲端 Jupyter Notebook 資源
下一篇
[Day 25] 使用Python Flask架設API吧!
系列文
全民瘋AI系列31

尚未有邦友留言

立即登入留言