Я обучил RandomForestClassifier из Python Sckit Learn Module с очень большим набором данных, но вопрос в том, как я могу сохранить эту модель и позволить другим людям применять ее на своем конце. Спасибо!
Как вывести RandomForest классификатор из python?
Ответ 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()