def write_out_plots_current_step(self, same_steps_only=True):
"""
This method will write out all plots available to the path
configured in self.lab_run_directory.
Parameters
---------
same_steps_only : boolean, optional
Write only if all experiment assistants in this lab assistant
are currently in the same step.
"""
step_string, same_step = self._compute_current_step_overall()
if same_steps_only and not same_step:
return
plot_base = os.path.join(self.lab_run_directory, "plots")
plot_step_base = os.path.join(plot_base, step_string)
ensure_directory_exists(plot_step_base)
plots_to_write = self.generate_all_plots()
#finally write out all plots created above to their files
for plot_name in plots_to_write.keys():
plot_fig = plots_to_write[plot_name]
write_plot_to_file(plot_fig, plot_name + "_" + step_string, plot_step_base)
def write_plots(self):
"""
Writes out the plots of this assistant.
"""
fig = self.plot_result_per_step()
filename = "result_per_step_%i" \
% len(self._experiment.candidates_finished)
path = self._experiment_directory_base + "/plots"
ensure_directory_exists(path)
write_plot_to_file(fig, filename, path)
write_plot_to_file(fig, "cur_state", self._experiment_directory_base)
plt.close(fig)
def write_out_plots_current_step(self, exp_ass=None, same_steps_only=True):
"""
This method will write out all plots available to the path
configured in self.lab_run_directory.
Parameters
---------
exp_ass : list, optional
List of experiment assistant names to include in the plots. Defaults to
None, which is equivalent to all.
same_steps_only : boolean, optional
Write only if all experiment assistants in this lab assistant
are currently in the same step.
"""
min_step = self._get_min_step()
if same_steps_only:
plot_up_to = min_step
else:
plot_up_to = None
plot_base = os.path.join(self._lab_run_directory, "plots")
plot_step_base = os.path.join(plot_base, "step_" + str(min_step))
ensure_directory_exists(plot_step_base)
if exp_ass is None:
exp_ass = self._exp_assistants.keys()
plots_to_write = self.generate_all_plots(exp_ass, plot_up_to)
#finally write out all plots created above to their files
for plot_name in plots_to_write.keys():
plot_fig = plots_to_write[plot_name]
write_plot_to_file(plot_fig, plot_name + "_step" + str(min_step),
plot_step_base)
plt.close(plot_fig)
