Commit 5931e6e6 authored by Andrei Roibu's avatar Andrei Roibu
Browse files

now also returning subject number in directory reader

parent a15b9d15
......@@ -64,7 +64,7 @@ def directory_reader(folder_location, subject_number=None, write_txt=False):
else:
break
return subDirectoryList
return subDirectoryList, number_of_subjects
def data_file_reader(data_file_path, folder_location, subject_number=None):
......@@ -117,7 +117,7 @@ def data_test_train_validation_split(data_folder_name, test_percentage, subject_
subDirectoryList = data_file_reader(
data_file, data_directory, subject_number)
else:
subDirectoryList = directory_reader(data_directory, subject_number)
subDirectoryList, _ = directory_reader(data_directory, subject_number)
subDirectoryList = np.array(subDirectoryList)
create_folder(data_folder_name)
......
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