diff options
| author |  Vijay Vasudevan <vrv@google.com> | 2017-02-17 17:05:49 -0800 |
|---|---|---|
| committer |  TensorFlower Gardener <gardener@tensorflow.org> | 2017-02-17 17:23:48 -0800 |
| commit | 93a975e114ee1c35f01ed3bdd47170e6f7129014 (patch) | |
| tree | e34255aff698fe6a4a586e7940337fd278947f58 /tensorflow/examples/learn/boston.py | |
| parent | eb9624017a0040e805fda622a5f9ec6681e24246 (diff) | |
Merge changes from github.
Change: 147897309
Diffstat (limited to 'tensorflow/examples/learn/boston.py')
| -rw-r--r-- | tensorflow/examples/learn/boston.py | 9 |
1 files changed, 5 insertions, 4 deletions
diff --git a/tensorflow/examples/learn/boston.py b/tensorflow/examples/learn/boston.py index 5d5ddff564..2986ff9106 100644 --- a/tensorflow/examples/learn/boston.py +++ b/tensorflow/examples/learn/boston.py @@ -43,11 +43,12 @@ def main(unused_argv): # Fit regressor.fit(x_train, y_train, steps=5000, batch_size=1) - + + # Transform + x_transformed = scaler.transform(x_test) + # Predict and score - y_predicted = list( - regressor.predict( - scaler.transform(x_test), as_iterable=True)) + y_predicted = list(regressor.predict(x_transformed, as_iterable=True)) score = metrics.mean_squared_error(y_predicted, y_test) print('MSE: {0:f}'.format(score)) |