Skip to content
Snippets Groups Projects

Modified the data_folder to be more generic

Merged Guillaume HEUSCH requested to merge generic_loader into master
1 file
+ 2
2
Compare changes
  • Side-by-side
  • Inline
@@ -64,11 +64,11 @@ def get_file_names_and_labels(files, data_folder, extension = ".hdf5", hldi_type
@@ -64,11 +64,11 @@ def get_file_names_and_labels(files, data_folder, extension = ".hdf5", hldi_type
if f.attack_type is None:
if f.attack_type is None:
label = 0
label = 1
else:
else:
label = 1
label = 0
file_name = os.path.join(data_folder, f.path + extension)
file_name = os.path.join(data_folder, f.path + extension)
Loading