2016-07-06 5 views
1
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 

Что лучшая практика для сохранитьобучена модель и использовать в другом месте?Лучшая практика «транспорт» обученная модель от sklearn

ответ

2

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 массивами

+0

Является ли это типом файла рассола? –

+0

Аналогично, но вы, вероятно, не можете десериализоваться с 'pickle', если вы сериализовались с помощью' joblib'. –

+0

спасибо !! работает отлично! –

 Смежные вопросы

  • Нет связанных вопросов^_^