Commit 7a578d22 authored by Amir MOHAMMADI's avatar Amir MOHAMMADI
Browse files

Make label and key list before repeating

parent b18d79de
......@@ -97,8 +97,8 @@ def bio_generator(database, preprocessor, groups, number_of_parallel_jobs,
for f, label, key in, labels, keys):
data = load_data(f, preprocessor, database)
if multiple_samples:
label = tf_repeat(label, len(data))
key = tf_repeat(key, len(data))
label = tf_repeat([label], len(data))
key = tf_repeat([key], len(data))
yield (data, label, key)
# load one data to get its type and shape
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment