перенести модель случайного леса sklearn на новый сервер

Question

Я построил одну модель с помощью sklearn RandomForestClassifier на старом сервере, и теперь мне нужно перенести ее на другой сервер. Как перенести модель на новый сервер? Какой пакет Python мне следует использовать? Соленый огурец? joblib? Спасибо!

Answer 1

pickle - это путь к go

from sklearn import model_selection
from sklearn.linear_model import LogisticRegression
import pickle

# Fit the model on training set
model = LogisticRegression()
model.fit(X_train, Y_train) # fit on some data ...

# save the model to disk
filename = 'finalized_model.sav'
pickle.dump(model, open(filename, 'wb'))


# load the model from disk
loaded_model = pickle.load(open(filename, 'rb'))
result = loaded_model.score(X_test, Y_test) # predict some test data
print(result)

Answer 2

Используйте «joblib». Предположим, ваша модель находится в переменной my_model. Тогда код 'joblib' будет go следующим образом:

# On your development machine
from joblib import dump
dump(my_model, 'model.joblib')

# On your new machine, following code would go to load the model
from joblib import load
my_model = load('model.joblib')

Примечание. Замените «model.joblib» на путь к файлу model.joblib.