【发布时间】:2020-04-29 09:59:24
【问题描述】:
我使用 scikit-learn 训练了一个模型,并使用 joblib 将其保存到磁盘。
joblib.dump(model, 'new_file_address')
我想知道是否可以将这个训练好的模型加载到 keras 中?
感谢您的关注。
【问题讨论】:
标签: python machine-learning keras scikit-learn
我使用 scikit-learn 训练了一个模型,并使用 joblib 将其保存到磁盘。
joblib.dump(model, 'new_file_address')
我想知道是否可以将这个训练好的模型加载到 keras 中?
感谢您的关注。
【问题讨论】:
标签: python machine-learning keras scikit-learn
您可以使用joblib 的load() 方法,并将保存模型的路径作为参数传递。这是相同的实现:
# Saving a model
joblib.dump(ml_model, 'path/to/saved_model.pkl')
# Loading a model
ml_model = joblib.load('path/to/saved_model.pkl')
【讨论】: