sklearn save trained model code example
Example 1: 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)
Example 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')