模型保存和调用的两种方式

模型的保存和调用

• Pickle
  • 导包
    • import pickle
  • 创建和训练模型
    • [外链图片转存失败,源站可能有防盗链机制,建议将图片保存下来直接上传(img-M3C9nWgF-1684802856365)(data/user-data/509837/images/283d4155c0b58cc84dc3a8163479ea25.png)]
  • 保存模型
    • pickle.dump(bst, open(“xgboostonboston.dat”,“wb”))
  • 查看保存路径
    • import sys
    • sys.path
  • 调用保存的模型
    • loaded_model = pickle.load(open(“xgboostonboston.dat”, “rb”))\
    • ypreds = loaded_model.predict(dtest)
• joblib
  • 导包
    • import joblib
  • 保存
    • joblib.dump(bst,“xgboost-boston.dat”)
    • bst必须是训练好的模型
  • 调用模型
    • loaded_model = joblib.load(“xgboost-boston.dat”)