个人技术分享

sklearn模型的保存和加载API

• from sklearn.externals import joblib
• 保存：joblib.dump(rf,‘test.pkl’)
• 加载：estimator =joblib.load(‘test.pkl’)

线性回归的模型保存加载案例

获取数据、划分数据集、标准化、预估器、保存模型、得出模型、模型评估

//保存模型
joblib.dump(estimator,"my_ridge.pkl")

//加载模型
estimator = ljoblib.load("my_ridge.pkl")