def save_plots()

in experiments/different_sketch_sizes_time_bar_plots.py [0:0]

• 40 lines of code
• 6 McCabe index (conditional complexity)

    def save_plots(self, exp_name, save_path):
        for i, sketch_size in enumerate(self.sketch_sizes):
            sketch_size_formula = self.sketch_size_formulas[i]
            for solver in self.solvers:
                self.plot_residuals(
                    solver,
                    sketch_size,
                    sketch_size_formula,
                    exp_name,
                    save_path=save_path,
                )

        # Plot time bar plot for each momentum version
        for solver in self.solvers:
            df_solver = self.times_df[self.times_df["solver"] == solver]
            fig_no_mom, fig_cst_mom, fig_inc_mom, fig_grouped = bar_plot_times_momentum(
                df_solver
            )

            if save_path:
                file_name = exp_name + "_time_bar"
                file_name = re.sub("[()]", "", file_name)

                full_path_no_mom = os.path.join(
                    save_path, f"{file_name}_{solver}_no_mom.pdf"
                )
                fig_no_mom.write_image(full_path_no_mom)

                full_path_cst_mom = os.path.join(
                    save_path, f"{file_name}_{solver}_cst_mom.pdf"
                )
                fig_cst_mom.write_image(full_path_cst_mom)

                full_path_inc_mom = os.path.join(
                    save_path, f"{file_name}_{solver}_inc_mom.pdf"
                )
                fig_inc_mom.write_image(full_path_inc_mom)

                full_path_grouped = os.path.join(
                    save_path, f"{file_name}_{solver}_grouped.pdf"
                )
                fig_grouped.write_image(full_path_grouped)
            else:
                fig_no_mom.show()
                fig_cst_mom.show()
                fig_inc_mom.show()
                fig_grouped.show()