机器学习-Python中训练模型的保存和再使用

模型保存

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)

猜你喜欢

转载自www.cnblogs.com/fuqia/p/9191634.html