change method to constructur

2018-05-29 15:35:14 +02:00 · 2018-05-29 15:35:14 +02:00 · a19fb1e39e
parent 2be5cdd2ce
commit a19fb1e39e
1 changed files with 1 additions and 1 deletions
--- a/pywatts/neural.py
+++ b/pywatts/neural.py
@ -5,7 +5,7 @@ class Net:
    regressor = None


-    def net(self, feature_cols):
+    def __init__(self, feature_cols):
        self.regressor = tf.estimator.DNNRegressor(feature_columns=feature_cols,
                                                   hidden_units=[50, 50],
                                                   model_dir='tf_pywatts_model')