在机器学习中,我们训练模型后,需要把模型保存在本地。利用joblib,实现简单的模型保存方式:
import joblib
#保存模型
def save_model(model, filepath):
# 后缀一般用pkl
joblib.dump(model, filename=filepath)
def load_model(filepath):
model = joblib.load(filepath)
return model
在机器学习中,我们训练模型后,需要把模型保存在本地。利用joblib,实现简单的模型保存方式:
import joblib
#保存模型
def save_model(model, filepath):
# 后缀一般用pkl
joblib.dump(model, filename=filepath)
def load_model(filepath):
model = joblib.load(filepath)
return model