机器学习,模型保存,模型加载
程序员文章站
2022-07-13 09:01:12
...
# 保存模型
from sklearn.externals import joblib
def save_model(mult_model, model_save_path):
"""
:param mult_model: 待保存的模型对象
:param model_save_path: 保存路径,例如 "./model.pkl"
:return: 没有返回值
"""
joblib.dump(mult_model, model_save_path)
# 加载模型
def load_model(model_path):
"""
:param model_path: 模型路径
:return: 返回加载后的模型对象
"""
return joblib.load(model_path)
下一篇: 使用网页版微信协议实现一个微信机器人