Python 模型保存和加载的两种方法：joblib库和pickle模块

• joblib库

# 安装包
!pip install joblib
# 导入包
import joblib
# 模型保存
joblib.dump(my_model, "mu_model.pkl")
# 模型加载
my_model_loaded = joblib.load("my_model.pkl")

• pickle模块
  • python2导入import cPickle
  • python3导入import pickle

# 导入包
import pickle
# 模型保存
pickle.dump(my_model, open('my_model.pkl','wb+')) # wb+，表示二进制写入
# 模型加载
my_model_loaded = pickle.load(open('my_model.pkl', 'rb')) # rb，表示二进制读取