2016-06-27 48 views
8

Sto cercando di imparare scikit-learn e Machine Learning utilizzando il set di dati Housing Boston.scikit-learn: come ridimensionare il risultato previsto 'y'

# I splitted the initial dataset ('housing_X' and 'housing_y') 
from sklearn.cross_validation import train_test_split 
X_train, X_test, y_train, y_test = train_test_split(housing_X, housing_y, test_size=0.25, random_state=33) 

# I scaled those two datasets 
from sklearn.preprocessing import StandardScaler 
scalerX = StandardScaler().fit(X_train) 
scalery = StandardScaler().fit(y_train) 
X_train = scalerX.transform(X_train) 
y_train = scalery.transform(y_train) 
X_test = scalerX.transform(X_test) 
y_test = scalery.transform(y_test) 

# I created the model 
from sklearn import linear_model 
clf_sgd = linear_model.SGDRegressor(loss='squared_loss', penalty=None, random_state=42) 
train_and_evaluate(clf_sgd,X_train,y_train) 

Sulla base di questo nuovo modello clf_sgd, sto cercando di prevedere il y in base alla prima istanza di X_train.

X_new_scaled = X_train[0] 
print (X_new_scaled) 
y_new = clf_sgd.predict(X_new_scaled) 
print (y_new) 

Tuttavia, il risultato è piuttosto strano per me (1.34032174, invece di 20-30, la gamma del prezzo delle case)

[-0.32076092 0.35553428 -1.00966618 -0.28784917 0.87716097 1.28834383 
    0.4759489 -0.83034371 -0.47659648 -0.81061061 -2.49222645 0.35062335 
-0.39859013] 
[ 1.34032174] 

Immagino che questo valore 1.34032174 dovrebbe essere ridimensionato, ma sto cercando di capire come farlo senza successo. Qualsiasi suggerimento è benvenuto Grazie mille.

+0

Non credo che è necessario applicare il ridimensionamento sul variabile di destinazione. Il ridimensionamento e altre tecniche di feature engineering sono applicate solo sui vettori di feature. –

risposta

11

È possibile utilizzare inverse_transform utilizzando l'oggetto scalery:

y_new_inverse = scalery.inverse_transform(y_new) 
+0

Grazie. Funziona. In effetti (e ovviamente), il risultato è del tutto identico al valore di "y_train". – Hookstark