From 5051a3922e44b40455fbed300fc1ac0b4422b137 Mon Sep 17 00:00:00 2001
From: Schneider Leo <leo.schneider@etu.ec-lyon.fr>
Date: Tue, 22 Apr 2025 11:49:28 +0200
Subject: [PATCH] add : ray result analysis
---
image_ref/main_ray.py | 4 +++-
1 file changed, 3 insertions(+), 1 deletion(-)
diff --git a/image_ref/main_ray.py b/image_ref/main_ray.py
index 0d678aa..80fa177 100644
--- a/image_ref/main_ray.py
+++ b/image_ref/main_ray.py
@@ -294,6 +294,8 @@ def analyze_ray_res(storage_path, exp_name,args, gpus_per_trial=1):
num_results = len(results)
print("Number of results:", num_results)
+ df = results.get_dataframe()
+ df.to_csv('df_results_contrastive.csv')
for i, result in enumerate(results):
if result.error:
print(f"Trial #{i} had an error:", result.error)
@@ -311,6 +313,6 @@ def analyze_ray_res(storage_path, exp_name,args, gpus_per_trial=1):
if __name__ == '__main__':
args = load_args_contrastive()
- analyze_ray_res()
+ analyze_ray_res('/lustre/fswork/projects/rech/bun/ucg81ws/these/pseudo_image/image_ref/ray_results/','ref_count_threshold_experiment',args)
# print(args)
# main(args)
--
GitLab