vanitasvitae/pywatts
1
0
Fork 0
Code Issues Pull requests Releases Wiki Activity

change method to constructur

Browse source
This commit is contained in:
Paul Schaub 2018-05-29 15:35:14 +02:00
parent 2be5cdd2ce
commit a19fb1e39e
Signed by: vanitasvitae
GPG key ID: 62BEE9264BF17311
1 changed files with 1 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

2
pywatts/neural.py
View file

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