Machine Learning With Python: Save And Load Trained Model

import pandas as pd
import numpy as np
from sklearn import linear_model

df = pd.read_csv("homeprices.csv")
df.head()

area price

0 2600 550000

1 3000 565000

2 3200 610000

3 3600 680000

4 4000 725000

model = linear_model.LinearRegression()
model.fit(df[['area']],df.price)

LinearRegression(copy_X=True, fit_intercept=True, n_jobs=None,

normalize=False)

model.coef_

array([135.78767123])

model.intercept_

180616.43835616432

model.predict([[5000]])

array([859554.79452055])

Save Model To a File Using Python Pickle

import pickle

with open('model_pickle','wb') as file:

    pickle.dump(model,file)

Load Saved Model

with open('model_pickle','rb') as file:

    mp = pickle.load(file)

mp.coef_

array([135.78767123])

mp.intercept_

180616.43835616432

mp.predict([[5000]])

array([859554.79452055])

Save Trained Model Using joblib

from sklearn.externals import joblib

joblib.dump(model, 'model_joblib')

['model_joblib']

Load Saved Model

mj = joblib.load('model_joblib')

mj.coef_

array([135.78767123])

mj.intercept_

180616.43835616432

mj.predict([[5000]])

array([859554.79452055])
Colab paid products
-
Cancel contracts here