Skip to content
Snippets Groups Projects
Commit 5b4b3962 authored by Pavel KORSHUNOV's avatar Pavel KORSHUNOV
Browse files

tests are passing for lstm network

parent 2867742c
No related branches found
No related tags found
No related merge requests found
Pipeline #
......@@ -79,7 +79,7 @@ def test_lstm_trainer_scratch():
validation_data = numpy.reshape(validation_data, (validation_data.shape[0], 28, 28, 1))
# Create scratch network
graph = scratch_lstm_network(train_data_shuffler, batch_size=validation_batch_size)
graph = scratch_lstm_network(train_data_shuffler, batch_size=batch_size)
# Setting the placeholders
embedding = Embedding(train_data_shuffler("data", from_queue=False), graph)
......@@ -102,7 +102,7 @@ def test_lstm_trainer_scratch():
trainer.train()
accuracy = validate_network(embedding, validation_data, validation_labels)
assert accuracy > 95
assert accuracy > 70
shutil.rmtree(directory)
del trainer
tf.reset_default_graph()
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment