Как сохранить 2 модели sklearn в одном файле

Question

На этой странице показаны методы сохранения модели с использованием pickle:

>>> import pickle
>>> s = pickle.dumps(clf)
>>> clf2 = pickle.loads(s)

или joblib:

>>> from sklearn.externals import joblib
>>> joblib.dump(clf, 'filename.joblib') 
>>> clf = joblib.load('filename.joblib') 

Что если нужносохранить 2 модели?Очевидно, я могу сохранить их в 2 отдельных файлах, но могу ли я сохранить их вместе в одном файле?Можно ли сохранить список с двумя моделями:

modlist = [clf1, clf2]

Как сохранить modlist в одном файле?

Answer 1

Вы можете выгружать / загружать различные структуры данных, используя pickle.В частности, одним из примеров может быть список моделей, оборудованных склеарном:

from sklearn import svm
from sklearn import datasets
from sklearn.naive_bayes import MultinomialNB
import pickle

clf1 = svm.SVC(gamma='scale')
clf2 = MultinomialNB()

iris = datasets.load_iris()
X, y = iris.data, iris.target
clf1.fit(X, y)  
clf2.fit(X, y)  

filename = "models.sav"

modlist = [clf1, clf2]

s = pickle.dump(modlist, open(filename, 'wb'))

modlist_loaded = pickle.load(open(filename, 'rb'))

clf1_loaded = modlist_loaded[0]
clf2_loaded = modlist_loaded[1]

print clf1_loaded.predict(X[:10])
print clf2_loaded.predict(X[:10])

Дайте мне знать, если это работает для вас.