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)
5
2
# 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')
Thank you!
2
0
Are there any code examples left?
New code examples in category Python