def make_images()

in vihds/xval.py [0:0]

• 60 lines of code
• 7 McCabe index (conditional complexity)

    def make_images(self):
        device_ids = list(range(len(self.settings.devices)))

        print("Making summary figure")
        f_summary = plotting.plot_prediction_summary(
            self.settings.devices,
            self.species_names,
            self.times,
            self.X_obs,
            self.iw_predict_mu,
            self.iw_predict_std,
            self.devices,
            "-",
        )
        self.save_figs(f_summary, "xval_fit")
        self.xval_writer.add_figure("Summary", f_summary, self.epoch)
        self.xval_writer.flush()

        if self.settings.separate_conditions is True:
            print("Making treatment figure")
            f_treatments = plotting.xval_treatments(self, device_ids)
            self.save_figs(f_treatments, "xval_treatments")
            self.xval_writer.add_figure("Treatment", f_treatments, self.epoch)
            self.xval_writer.flush()

        print("Making species figure")
        f_species = plotting.species_summary(
            self.species_names, self.treatments, self.devices, self.times, self.iw_states, device_ids, self.settings,
        )
        self.save_figs(f_species, "xval_species")
        self.xval_writer.add_figure("Species", f_species, self.epoch)
        self.xval_writer.flush()

        print("Making global parameters figure")
        f_gparas = plotting.xval_global_parameters(self)
        if f_gparas is not None:
            self.save_figs(f_gparas, "xval_global_parameters")
            self.xval_writer.add_figure("Parameters/Globals", f_gparas, self.epoch)
            self.xval_writer.flush()

        print("Making variable parameters figure")
        f_vparas = plotting.xval_variable_parameters(self)
        if f_vparas is not None:
            self.save_figs(f_vparas, "xval_variable_parameters")
            self.xval_writer.add_figure("Parameters/Variable", f_vparas, self.epoch)
            self.xval_writer.flush()

        print("Making summary device figures")
        for u in device_ids:
            print("- %s" % self.settings.pretty_devices[u])
            device = self.settings.devices[u]
            f_summary_i = plotting.xval_fit_summary(self, u, separatedInputs=self.settings.separate_conditions)
            self.save_figs(f_summary_i, "xval_summary_%s" % device)
            self.xval_writer.add_figure("Device_Summary/" + device, f_summary_i, self.epoch)
        self.xval_writer.flush()

        print("Making individual device figures")
        for u in device_ids:
            print("- %s" % self.settings.pretty_devices[u])
            device = self.settings.devices[u]
            if self.settings.separate_conditions is True:
                # TODO: Check just 1 treatment? 2treatments function fails when there is just 1 treatment
                f_indiv_i = plotting.xval_individual_2treatments(self, u)
            else:
                f_indiv_i = plotting.xval_individual(self, u)
            self.save_figs(f_indiv_i, "xval_individual_%s" % device)
            self.xval_writer.add_figure("Device_Individual/" + device, f_indiv_i, self.epoch)
        self.xval_writer.flush()