Skip to content

Commit

Permalink
Comparison notebook: fix paths and time_start var in supp script
Browse files Browse the repository at this point in the history
  • Loading branch information
L-in-da authored Mar 7, 2024
1 parent 84c77e0 commit 576b115
Showing 1 changed file with 4 additions and 4 deletions.
Original file line number Diff line number Diff line change
Expand Up @@ -180,9 +180,9 @@
" for dc, data_comp in enumerate([dataset_intra, dataset_inter]):\n",
" dataset2 = data_comp\n",
" assert dataset1.shape == dataset2.shape, f\"Dataset shapes do not match: {dataset1.shape} vs. {dataset2.shape}\"\n",
" time_start = time.time()\n",
" if mapping[exp_name] == 'MMD':\n",
" print(f'MMD {cfg.data} {d} {cfg.mmd_bandwidth[d]}')\n",
" time_start = time.time()\n",
" output = experiment.run_experiment(dataset1=dataset1,\n",
" dataset2=dataset2,\n",
" dataset_size = cfg.n[0],\n",
Expand Down Expand Up @@ -231,10 +231,10 @@
"axes[-1, 2].set_xlabel('dimensions')\n",
"\n",
"fig.tight_layout()\n",
"fig.savefig(f\"../../plots/metric_comparison_sim_budget_{cfg.n[0]}_dim_size_{cfg.data[0]}_adapted_bandwidth.png\", dpi=300)\n",
"fig.savefig(f\"../../plots/metric_comparison_sim_budget_{cfg.n[0]}_dim_size_{cfg.data[0]}_adapted_bandwidth.pdf\", dpi=300)\n",
"fig.savefig(f\"./results/plots/{cfg.exp_log_name}_metric_comparison_sim_budget_{cfg.n[0]}_dim_size_{cfg.data[0]}_adapted_bandwidth.png\", dpi=300)\n",
"fig.savefig(f\"./results/plots/{cfg.exp_log_name}_metric_comparison_sim_budget_{cfg.n[0]}_dim_size_{cfg.data[0]}_adapted_bandwidth.pdf\", dpi=300)\n",
"\n",
"print(f\"Plots saved to {cfg.data[0]}.png\")\n",
"print(f\"Plots saved")\n",
"print(\"Finished running experiments.\")\n"
]
}
Expand Down

0 comments on commit 576b115

Please sign in to comment.