1. 程式人生 > >機器學習-Python中訓練模型的保存和再使用

機器學習-Python中訓練模型的保存和再使用

ext model job 模型保存 PE dir code pre del

模型保存

BP:model.save(save_dir)

SVM:

from sklearn.externals import joblib
joblib.dump(clf, save_dir)

模型調用:

BP:

from keras.models import load_model
model = load_model(open_dir)

SVM:

from sklearn.externals import joblib
model = joblib.load(open_dir)

機器學習-Python中訓練模型的保存和再使用