1. 程式人生 > >模型儲存,載入與呼叫

模型儲存,載入與呼叫

模型儲存

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)

模型呼叫

result = model.predict(x_predict)

在這裡插入圖片描述