Merge pull request #2994 from Jendker/patch-1

Bug fix - test_csvs argument was ignored
This commit is contained in:
Kelly Davis 2020-05-15 19:33:48 +02:00 committed by GitHub
commit 3b4c39f80d
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

View File

@ -50,7 +50,6 @@ def evaluate(test_csvs, create_model):
else:
scorer = None
test_csvs = FLAGS.test_files.split(',')
test_sets = [create_dataset([csv], batch_size=FLAGS.test_batch_size, train_phase=False) for csv in test_csvs]
iterator = tfv1.data.Iterator.from_structure(tfv1.data.get_output_types(test_sets[0]),
tfv1.data.get_output_shapes(test_sets[0]),