Commit 8decf47f authored by Vojtěch Řehák's avatar Vojtěch Řehák
Browse files

bugfix - plots into plots_dir

parent 9e2150fd

writers.py

+2 −2
Original line number Original line Diff line number Diff line
@@ -97,9 +97,9 @@ class DiskExperimentWriter(DiskWriter): 
        if self.plot:

        if self.plot:

            plots = {key: key for key in self.data['runs_df']['plots'].columns}

            plots = {key: key for key in self.data['runs_df']['plots'].columns}

            plot_values(self.data['runs_df']['plots'].reset_index(),

            plot_values(self.data['runs_df']['plots'].reset_index(),

                        os.path.join(self.results_dir, 'training_progress.pdf'), plots=plots)

                        os.path.join(self.plots_dir, 'training_progress.pdf'), plots=plots)

            plot_times(self.data['runs_df']['times'].reset_index(),

            plot_times(self.data['runs_df']['times'].reset_index(),

                       os.path.join(self.results_dir, 'avg_times.pdf'))

                       os.path.join(self.plots_dir, 'avg_times.pdf'))





        super(DiskExperimentWriter, self).close()

        super(DiskExperimentWriter, self).close()

        return self.data['info']

        return self.data['info']