graphorge.gnn_base_model.predict.prediction_plots.plot_prediction_loss_history

plot_prediction_loss_history(loss_history, loss_type=None, is_log_loss=False, loss_scale='linear', filename='prediction_loss_history', save_dir=None, is_save_fig=False, is_stdout_display=False, is_latex=False)[source]

Plot model prediction process loss history.

Parameters:
  • loss_history (dict) – One or more prediction processes loss histories, where each loss history (key, str) is stored as a list of prediction steps loss values (item, list). Dictionary keys are taken as labels for the corresponding prediction processes loss histories.

  • loss_type (str, default=None) – Loss type. If provided, then loss type is added to the y-axis label.

  • is_log_loss (bool, default=False) – Applies logarithm to loss values if True, keeps original loss values otherwise.

  • loss_scale ({'linear', 'log'}, default='linear') – Loss axis scale type.

  • filename (str, default='prediction_loss_history') – Figure name.

  • save_dir (str, default=None) – Directory where figure is saved. If None, then figure is saved in current working directory.

  • is_save_fig (bool, default=False) – Save figure.

  • is_stdout_display (bool, default=False) – True if displaying figure to standard output device, False otherwise.

  • is_latex (bool, default=False) – If True, then render all strings in LaTeX. If LaTex is not available, then this option is silently set to False and all input strings are processed to remove $(…)$ enclosure.