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