modifying benchmarks

Author: quentinhaenn

examples/plot_benchmark_custom.py

@@ -184,16 +184,29 @@ def benchmark_radius_clustering():
 fig.suptitle("Benchmark of Radius Clustering Solvers", fontsize=16)
 
 axs['time'].set_yscale('log')  # Use logarithmic scale for better visibility
-for algo, algo_results in results.items():
+#for algo, algo_results in results.items():
     # Plot execution time
-    axs['time'].plot(
-        DATASETS.keys(),
-        algo_results["time"],
-        marker='o',
-        label=algo,
-    )
+#    axs['time'].plot(
+#        DATASETS.keys(),
+#        algo_results["time"],
+#        marker='o',
+#        label=algo,
+#    )
     # Plot number of clusters
 
+algorithms = list(results.keys())
+dataset_names = list(DATASETS.keys())
+n_algos = len(algorithms)
+x_indices = np.arange(len(dataset_names))  # the label locations
+bar_width = 0.8 / n_algos  # the width of the bars, with some padding
+
+for i, algo in enumerate(algorithms):
+    times = results[algo]["time"]
+    # Calculate position for each bar in the group to center them
+    position = x_indices - (n_algos * bar_width / 2) + (i * bar_width) + bar_width / 2
+    axs['time'].bar(position, times, bar_width, label=algo)
+# --- End of change ---
+
 for i, (name, (dataset, _)) in enumerate(DATASETS.items()):
     axs[name].bar(
         results.keys(),
@@ -207,14 +220,15 @@ def benchmark_radius_clustering():
         linestyle='--',
     )
     axs[name].set_title(name)
-    axs[name].set_xlabel("Algorithms")
 
 axs["iris"].set_ylabel("Number of clusters")
 axs["glass"].set_ylabel("Number of clusters")
 
 axs['time'].set_title("Execution Time (log scale)")
 axs['time'].set_xlabel("Datasets")
 axs['time'].set_ylabel("Time (seconds)")
+axs['time'].set_xticks(x_indices)  # Set tick positions to be at the center of the groups
+axs['time'].set_xticklabels(dataset_names)
 axs['time'].legend(title="Algorithms")
 plt.tight_layout()
 plt.show()