Predicting new data using sklearn after standardizing the training data

Question

I am using Sklearn to build a linear regression model (or any other model) with the following steps:

X_train and Y_train are the training data

1. Standardize the training data

     X_train = preprocessing.scale(X_train)
   

2. fit the model

    model.fit(X_train, Y_train)
   

Once the model is fit with scaled data, how can I predict with new data (either one or more data points at a time) using the fit model?

What I am using is

1. Scale the data

   NewData_Scaled = preprocessing.scale(NewData)
   

2. Predict the data

   PredictedTarget = model.predict(NewData_Scaled)
   

I think I am missing a transformation function with preprocessing.scale so that I can save it with the trained model and then apply it on the new unseen data? any help please.

Answer 1

Take a look at these docs.

You can use the StandardScaler class of the preprocessing module to remember the scaling of your training data so you can apply it to future values.

from sklearn.preprocessing import StandardScaler
X_train = np.array([[ 1., -1.,  2.],
                    [ 2.,  0.,  0.],
                    [ 0.,  1., -1.]])
scaler = StandardScaler().fit(X_train)

scaler has calculated the mean and scaling factor to standardize each feature.

>>>scaler.mean_
array([ 1. ...,  0. ...,  0.33...])
>>>scaler.scale_                                       
array([ 0.81...,  0.81...,  1.24...])

To apply it to a dataset:

import numpy as np

X_train_scaled = scaler.transform(X_train)
new_data = np.array([-1.,  1., 0.])    
new_data_scaled = scaler.transform(new_data)
>>>new_data_scaled
array([[-2.44...,  1.22..., -0.26...]])