python训练好的模型保存与加载
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
用户评论