Skip to content

Commit 7f927fc

Browse files
committed
inlined demos and created fake mini images
1 parent b5f199c commit 7f927fc

5 files changed

Lines changed: 989 additions & 988 deletions

File tree

LearnDigitz/LearnDigitz.py

Lines changed: 5 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -316,17 +316,18 @@ def main(_):
316316
# 0-9 digits recognition => 10 classes
317317
y = tf.placeholder(tf.float32, [None, 10], name='label')
318318

319-
# model init=tf.keras.initializers.he_uniform()
320-
predictor = linear_model(x)
319+
# model
320+
predictor = multilayer_perceptron_relu_softmax_model(x,
321+
init=tf.keras.initializers.he_uniform())
321322

322323
# model accuracy
323324
accuracy = get_accuracy(predictor, y)
324325

325326
# cost / loss
326-
cost = squared_error_loss(predictor, y)
327+
cost = cross_entropy_loss(predictor, y)
327328

328329
# optimizer
329-
optimizer = sgd_optimizer(cost, learning_rate)
330+
optimizer = adam_optimizer(cost, learning_rate)
330331

331332
# training
332333
train_model(optimizer, cost, accuracy, x, y, batch_size, training_epochs)

0 commit comments

Comments
 (0)