def save_plots()

in autoencoder.py [0:0]

• 21 lines of code
• 5 McCabe index (conditional complexity)

    def save_plots(self, path):

        for train_set in [True, False]:
            set_name = "train" if train_set else "test"

            x1_plot_path = os.path.join(path, f"x1_{set_name}_reconstructions")
            self.plot_x1_reconstructions(save_name=x1_plot_path, train_set=train_set)

            # store x2 reconstructions only when using supervised latent operator
            if self.use_latent_op:
                x2_plot_path = os.path.join(path, f"x2_{set_name}_reconstructions")
                self.plot_x2_reconstructions(
                    save_name=x2_plot_path, train_set=train_set
                )

            transformation_name = (
                "translations"
                if self.transformation_param_name != "angle"
                else "rotations"
            )
            multiple_rotations_path = os.path.join(
                path, f"x_{set_name}_{transformation_name}"
            )
            self.plot_multiple_rotations(
                save_name=multiple_rotations_path, train_set=train_set
            )