Skip to content
Snippets Groups Projects
Commit 5051a392 authored by Schneider Leo's avatar Schneider Leo
Browse files

add : ray result analysis

parent 82a42976
No related branches found
No related tags found
No related merge requests found
......@@ -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)
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment