Update dnn_housing.py

TortoiseHam · Jun 18, 2019 · a46c519 · a46c519
1 parent c864479
commit a46c519
Showing 1 changed file with 1 addition and 2 deletions.
diff --git a/tabular/dnn_housing.py b/tabular/dnn_housing.py
@@ -1,4 +1,3 @@
-# from fastestimator.pipeline.static.preprocess import Reshape
 from fastestimator.estimator.estimator import Estimator
 from fastestimator.pipeline.pipeline import Pipeline
 from sklearn.preprocessing import StandardScaler
@@ -53,4 +52,4 @@ def get_estimator(epochs=30, batch_size=32):
                           pipeline=pipeline,
                           epochs= epochs,
                           log_steps=10)
-    return estimator
+    return estimator