def log_eval_plots(self, tag, y_true, y_prob, step=None):
    y_true = y_true.detach().cpu()
    y_prob = y_prob.detach().cpu()
    step = (step or self.global_step)
    reliability = utils.draw_reliability_plot(y_true, y_prob, n_bins=10)
    mlflow.log_figure(reliability, (tag + f'/reliability/{step}.png'))
    reliability = utils.fig2rgb(reliability)
    self.logger.experiment.add_image((tag + '/reliability'), reliability, step)
    roc = utils.draw_roc(y_true, y_prob)
    mlflow.log_figure(roc, (tag + f'/roc/{step}.png'))
    roc = utils.fig2rgb(roc)
    self.logger.experiment.add_image((tag + '/roc'), roc, step)
    ssp = utils.draw_ssp(y_true, y_prob)
    mlflow.log_figure(ssp, (tag + f'/ssp/{step}.png'))
    ssp = utils.fig2rgb(ssp)
    self.logger.experiment.add_image((tag + '/ssp'), ssp, step)
def log_eval_plots(self, tag, y_true, y_prob, step=None):
    y_true = y_true.detach().cpu()
    y_prob = y_prob.detach().cpu()
    step = (step or self.global_step)
    reliability = utils.draw_reliability_plot(y_true, y_prob, n_bins=10)
    mlflow.log_figure(reliability, (tag + f'/reliability/{step}.png'))
    reliability = utils.fig2rgb(reliability)
    self.logger.experiment.add_image((tag + '/reliability'), reliability, step)
    roc = utils.draw_roc(y_true, y_prob)
    mlflow.log_figure(roc, (tag + f'/roc/{step}.png'))
    roc = utils.fig2rgb(roc)
    self.logger.experiment.add_image((tag + '/roc'), roc, step)
    ssp = utils.draw_ssp(y_true, y_prob)
    mlflow.log_figure(ssp, (tag + f'/ssp/{step}.png'))
    ssp = utils.fig2rgb(ssp)
    self.logger.experiment.add_image((tag + '/ssp'), ssp, step)