Skip to content

Commit

Permalink
constructed cost func and optimizer
Browse files Browse the repository at this point in the history
  • Loading branch information
LordSomen committed Oct 14, 2018
1 parent 297181b commit 4056698
Showing 1 changed file with 36 additions and 0 deletions.
36 changes: 36 additions & 0 deletions AAN/arti_neural_net.py
Original file line number Diff line number Diff line change
Expand Up @@ -89,4 +89,40 @@ def neuron_layer(X,n_neurons,name,activation=None):

#%%

with tf.name_scope("dnn"):
hidden1 = neuron_layer(X , n_hidden1 , name="hidden1" ,
activation=tf.nn.relu)
hidden2 = neuron_layer(hidden1,n_hidden2,name="hidden2",
activation=tf.nn.relu)
logits = neuron_layer(hidden2,n_outputs,name="outputs")

#%%

with tf.name_scope("loss"):
xentropy = tf.nn.sparse_softmax_cross_entropy_with_logits(labels=
Y,logits=logits)
loss = tf.reduce_mean(xentropy,name="loss")


#%%

learning_rate = 0.01

with tf.name_scope("train"):
optimizer = tf.train.GradientDescentOptimizer(learning_rate)
training_op = optimizer.minimize(loss)

#%%
with tf.name_scope("eval"):
correct = tf.nn.in_top_k(logits,Y,1)
accuracy = tf.reduce_mean(tf.cast(correct,tf.float32))

#%%

init = tf.global_variables_initializer()
saver = tf.train.Saver()

#%%



0 comments on commit 4056698

Please sign in to comment.