- 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,表示二进制读取