summaryrefslogtreecommitdiff
path: root/boston.py
diff options
context:
space:
mode:
authorschneefux <schneefux+commit@schneefux.xyz>2016-02-21 18:07:05 +0100
committerschneefux <schneefux+commit@schneefux.xyz>2016-02-21 18:07:05 +0100
commit56eeb05b4394c9c420932bea254dcfc813092d45 (patch)
tree88a0812e8882bd61786cc4d375e36fbc4bb567eb /boston.py
parentd92ed343fef98c4cfbf2086b6f4b1a5477da8fca (diff)
downloadboston-neuralnet-56eeb05b4394c9c420932bea254dcfc813092d45.tar.gz
boston-neuralnet-56eeb05b4394c9c420932bea254dcfc813092d45.zip
optimiere Konfiguration
Diffstat (limited to 'boston.py')
-rw-r--r--boston.py10
1 files changed, 5 insertions, 5 deletions
diff --git a/boston.py b/boston.py
index 7968dec..e8382de 100644
--- a/boston.py
+++ b/boston.py
@@ -23,13 +23,13 @@ x_train, x_test, y_train, y_test = train_test_split(
)
BATCH = 50
-STEPS = 100 # epochs
-HIDDEN = 9 # hidden nodes in 1 layer
+STEPS = 300 # epochs
+HIDDEN = 11 # hidden nodes in 1 layer
IN = 13 # input nodes
OUT = 1 # output node(s)
-LEARNINGRATE = 0.1
-DECAYRATE = 0.95
-DECAYSTEPS = 10
+LEARNINGRATE = 0.01
+DECAYRATE = 0.96
+DECAYSTEPS = 50
RUNNAME = 'B'+str(BATCH)+'S'+str(STEPS)+'H'+str(HIDDEN)+'LR'+str(LEARNINGRATE)\
+ 'DR'+str(DECAYRATE)+'DS'+str(DECAYSTEPS)+'SD'+str(SEED)