Как вывести RandomForest классификатор из python?

Я обучил RandomForestClassifier из Python Sckit Learn Module с очень большим набором данных, но вопрос в том, как я могу сохранить эту модель и позволить другим людям применять ее на своем конце. Спасибо!

Ответ 1

Рекомендуемый метод - использовать joblib, это приведет к получению файла намного меньшего размера, чем маринада:

from sklearn.externals import joblib
joblib.dump(clf, 'filename.pkl') 

#then your colleagues can load it

clf = joblib.load('filename.pkl')

Посмотреть онлайн документы

Ответ 2

Пробовали ли вы травить RandomForestClassifier с помощью модуля Pickle, а затем сохраняете его на диске?

Вот пример, основанный на pickle docs:

import pickle

classifier = RandomForestClassifier(etc)
output = open('classifier.pkl', 'wb')
pickle.dump(classifier, output)
output.close()

Затем "другие люди" могут перезагрузить маринованный объект следующим образом:

import pickle

f = open('classifier.pkl', 'rb')
classifier = pickle.load(f)
f.close()