Add zero div check

experiments/scripts/attack.py

@@ -157,8 +157,8 @@ def data_saver(queue_in, queue_log, queue_recurse, output_file,
                     "attacks_succeeded": ch_suc,
                     "attacks_all": ch_all,
                     "synonyms_nbr": synonyms_nbr,
-                    "success_rate": ch_suc / ch_all,
-                    "success_rate_per_synonym": ch_suc / synonyms_nbr,
+                    "success_rate": ch_suc / ch_all if ch_all > 0 else 0,
+                    "success_rate_per_synonym": ch_suc / synonyms_nbr if synonyms_nbr > 0 else 0,
                     "time": time() - start,
                     "samples": samples,
                     "samples_succ": samples_succ,
@@ -176,8 +176,8 @@ def data_saver(queue_in, queue_log, queue_recurse, output_file,
             "attacks_succeeded": ch_suc,
             "attacks_all": ch_all,
             "synonyms_nbr": synonyms_nbr,
-            "success_rate": ch_suc / ch_all,
-            "success_rate_per_synonym": ch_suc / synonyms_nbr,
+            "success_rate": ch_suc / ch_all if ch_all > 0 else 0,
+            "success_rate_per_synonym": ch_suc / synonyms_nbr if synonyms_nbr > 0 else 0,
             "time": time() - start,
             "samples": samples,
             "samples_succ": samples_succ,
@@ -191,7 +191,7 @@ def data_saver(queue_in, queue_log, queue_recurse, output_file,
     except Exception as e:
         queue_log.put(f"Error in data saver: {e}")
         with open(log_file, "a") as f:
-            f.write("Saver failed with {e}\n")
+            f.write(f"Saver failed with {e}\n")
         queue_in.put(None)
 
 
@@ -336,7 +336,7 @@ def main(dataset_name: str, attack_type: str):
     output_path = os.path.join(output_dir, "test.jsonl")
     dataset_df = pd.read_json(input_file, lines=True)
 
-    log_file = f"{attack_type}_{dataset_name}_{datetime.now().strftime('%Y-%m-%d_%H-%M')}.log"
+    log_file = f"logs/{attack_type}_{dataset_name}_{datetime.now().strftime('%Y-%m-%d_%H-%M')}.log"
 
     m = Manager()
     queues = [m.Queue(maxsize=QUEUE_SIZE) for _ in range(4)]