wassname/ray
1
0
Fork 0
mirror of https://github.com/wassname/ray.git synced 2026-08-20 12:40:44 +08:00
Code Issues Packages Projects Releases Wiki Activity

clean up imports (#230)

Browse Source
This commit is contained in:
Robert Nishihara
2016-07-08 12:46:47 -07:00
committed by Philipp Moritz
parent 191909dd93
commit 5dd411546d
16 changed files with 81 additions and 105 deletions
Download Patch File Download Diff File Expand all files Collapse all files
examples/hyperopt/functions.py
+3 -3
View File
@@ -38,12 +38,12 @@ def train_cnn(params, epochs):
sess.run(train_step, feed_dict={x: batch[0], y: batch[1], keep_prob: keep})
if i % 100 == 0: # checks if accuracy is low enough to stop early every set number of epochs
train_ac = accuracy.eval(feed_dict={x: batch[0], y: batch[1], keep_prob: 1.0})
if train_ac < 0.25: # Accuracy threshold is on a application to application basis.
if train_ac < 0.25: # Accuracy threshold is on a application to application basis.
totalacc = accuracy.eval(feed_dict={x: mnist.validation.images, y: mnist.validation.labels, keep_prob: 1.0})
return totalacc
totalacc = accuracy.eval(feed_dict={x: mnist.validation.images, y: mnist.validation.labels, keep_prob: 1.0})
return totalacc
return totalacc.astype("float64")
def cnn_setup(x, y, keep_prob, lr, stddev):
first_hidden = 32
second_hidden = 64