＆＃34;运输的最佳实践＆＃34;来自sklearn的训练模型

Question

import matplotlib.pyplot as plt
import numpy as np
from sklearn import datasets, linear_model


# Create linear regression object
regr = linear_model.LinearRegression()

# Train the model using the training sets
regr.fit(X_train, y_train)
# how save ?????
# save here

最佳做法 保存 经过培训的模型并在其他地方使用？

Answer 1

sklearn有一个joblib模块，用于保存模型和/或保存到文件中：

from sklearn.externals import joblib

joblib.dump(regr, 'file_name.pkl')

# load pickled model later
regr = joblib.load('file_name.pkl') 

你也可以使用Python的内置pickle，但docs建议使用joblib来有效地挑选具有大numpy数组的对象