Merge pull request #2994 from Jendker/patch-1
Bug fix - test_csvs argument was ignored
This commit is contained in:
commit
3b4c39f80d
@ -50,7 +50,6 @@ def evaluate(test_csvs, create_model):
|
|||||||
else:
|
else:
|
||||||
scorer = None
|
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]
|
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]),
|
iterator = tfv1.data.Iterator.from_structure(tfv1.data.get_output_types(test_sets[0]),
|
||||||
tfv1.data.get_output_shapes(test_sets[0]),
|
tfv1.data.get_output_shapes(test_sets[0]),
|
||||||
|
Loading…
x
Reference in New Issue
Block a user