Python一种保存模型的简单方式:用joblib保存训练模型

在机器学习中,我们训练模型后,需要把模型保存在本地。利用joblib,实现简单的模型保存方式:

import joblib

#保存模型
def save_model(model, filepath):
    # 后缀一般用pkl
	joblib.dump(model, filename=filepath)

def load_model(filepath):
	model = joblib.load(filepath)
	return model

猜你喜欢

转载自blog.csdn.net/guyu1003/article/details/109017083