模型的保存和加载

API

如果我们希望保存和加载模型，我们可以使用 joblib 库，它可以将 Python 对象序列化到磁盘，并在需要时重新加载它们。

sklearn 已经为我们封装好了 joblib 的接口，我们只需要调用 dump 和 load 方法即可：

from sklearn.externals import joblib

# ... 训练模型 ...

# 保存模型
joblib.dump(model, 'filename.pkl')

# 加载模型
model = joblib.load('filename.pkl')