Andrea Maldonado commited on
Commit
bc9e7f8
·
1 Parent(s): d86aae0

Summarize names of genELs with over 10 features

Browse files
Files changed (2) hide show
  1. gedi/generator.py +1 -2
  2. gedi/utils/io_helpers.py +1 -1
gedi/generator.py CHANGED
@@ -192,8 +192,7 @@ class GenerateEventLogs():
192
  try:
193
  identifier = [x for x in task[1] if isinstance(x, str)][0]
194
  except IndexError:
195
- identifier = task[0]+1
196
- #task = task[1].loc[lambda x, identifier=identifier: x!=identifier]
197
  task = task[1].drop('log', errors='ignore')
198
  self.objectives = task.dropna().to_dict()
199
  random.seed(RANDOM_SEED)
 
192
  try:
193
  identifier = [x for x in task[1] if isinstance(x, str)][0]
194
  except IndexError:
195
+ identifier = "genEL"+str(task[0]+1)
 
196
  task = task[1].drop('log', errors='ignore')
197
  self.objectives = task.dropna().to_dict()
198
  random.seed(RANDOM_SEED)
gedi/utils/io_helpers.py CHANGED
@@ -60,7 +60,7 @@ def get_output_key_value_location(obj, output_path, identifier, obj_keys=None):
60
 
61
  obj_values = [round(x, 4) for x in [*obj_sorted.values()]]
62
 
63
- if len(obj_keys) > 7:
64
  folder_path = os.path.join(output_path, f"{len(obj_keys)}_features")
65
  generated_file_name = f"{identifier}"
66
  else:
 
60
 
61
  obj_values = [round(x, 4) for x in [*obj_sorted.values()]]
62
 
63
+ if len(obj_keys) > 10:
64
  folder_path = os.path.join(output_path, f"{len(obj_keys)}_features")
65
  generated_file_name = f"{identifier}"
66
  else: