From a19fb1e39e8997731ea8731cbd5e726384e33d92 Mon Sep 17 00:00:00 2001 From: Paul Schaub Date: Tue, 29 May 2018 15:35:14 +0200 Subject: [PATCH] change method to constructur --- pywatts/neural.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/pywatts/neural.py b/pywatts/neural.py index 28474c3..18ad9de 100644 --- 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')