Skip to content

Commit 8bd0776

Browse files
mheilmanVijay Vasudevan
authored andcommitted
fix scaler transform boston skflow example (tensorflow#2152)
fit_transform was being used instead of transform for the test set in the boston example
1 parent 35a9349 commit 8bd0776

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

tensorflow/examples/skflow/boston.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -40,6 +40,6 @@
4040
regressor.fit(X_train, y_train)
4141

4242
# Predict and score
43-
score = metrics.mean_squared_error(regressor.predict(scaler.fit_transform(X_test)), y_test)
43+
score = metrics.mean_squared_error(regressor.predict(scaler.transform(X_test)), y_test)
4444

4545
print('MSE: {0:f}'.format(score))

0 commit comments

Comments
 (0)