Answers for "how to save sklearn model"

3

save machine learning model

# fit the model
model.fit(X_train, y_train)

# save the model
import pickle
pickle.dump(model, open("model.pkl", "wb"))

# load the model
model = pickle.load(open("model.pkl", "rb"))

# use model to predict
y_pred = model.predict(X_input)
Posted by: Guest on July-24-2020
7

save and load sklearn model PKL

model.fit(X_train, Y_train)
# 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)
Posted by: Guest on March-06-2020
2

sklearn save model

# save as object
import pickle

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

# save to file
from joblib import dump

dump(clf, 'filename.joblib') 

# load from file
from joblib import load

clf = load('filename.joblib')
Posted by: Guest on November-26-2020

Python Answers by Framework

Browse Popular Code Answers by Language