從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但是文檔建議使用joblib來有效地對具有大型numpy數組的對象進行酸洗。