Python es una forma sencilla de guardar el modelo: guarde el modelo de entrenamiento con joblib

 

En el aprendizaje automático, después de entrenar el modelo, necesitamos guardar el modelo localmente. Use joblib para implementar un método simple de guardado de modelos:

import joblib

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

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

 

Supongo que te gusta

Origin blog.csdn.net/guyu1003/article/details/109017083
Recomendado
Clasificación