上传者: 38708223
|
上传时间: 2021-07-12 23:35:48
|
文件大小: 37KB
|
文件类型: PDF
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