python-scikit-learnHow to save ML model to file
import pickle
from sklearn import datasets, linear_model, model_selection, metrics
X, y = datasets.load_diabetes(return_X_y=True)
X_train, X_test, y_train, y_test = model_selection.train_test_split(X, y)
model = linear_model.LinearRegression()
model.fit(X_train, y_train)
pickle.dump(model, open('/tmp/model.ml', 'wb'))ctrl + cpickle.dump(saves given model to the given file | /tmp/model.mlpath to the file to save model to |
open(opens file descriptor (to write data to it) | |
More of Python Scikit Learn
- Use imputer for one column only
- How to get best parameters from grid search CV
- How to get best estimator from grid search CV
- Grid search CV example
- PCA dimensionality reduction example
- How to inverse transform PCA reduced data
- Neural network example
- Feature selection using KBest and chi2
- Xgboost regression usage example
- How to use n_jobs for parallel training
See more codes...