python训练好的模型保存与加载
当我们训练好一个模型model后,如果在其他程序中或者下次想继续使用这个model,我们就需要把这个model保存下来,下次使用时直接导入就好了,不需要重新训练。
方式一:
采用joblib模块来保存model,首先安装joblib:
pip install joblib
核心两行代码如下
# 保存 model
joblib.dump(regr, '../../model/regr.pkl')
# 加载 model
clf = joblib.load('../../model/regr.pkl')
完整示例代码如下:
#!/usr/bin/env pyt
1