Visualization Modules

Loss Visuals

visualization.loss_visuals.visualize_loss(data, title, xlabel, ylabel, savefig=False, path=None, img_name=None)

Creates overlapping line plots for multiple sources of losses.

Parameters:

data (list or matrix) – Losses to plot in a line plot, different sources of losses must be in different lists or rows
title (str) – Title of plot
xlabel (str) – x-axis label of plot
ylabel (str) – y-axis label of plot
savefig (bool, optional) – If True, saves a copy of the graph in a png file, must also specify path and img_name
path (str, optional) – Path to store saved plot image in, doesn’t work if savefig = False
img_name (str, optional) – Name of saved plot image, doesn’t work if savefig = False

Returns:

None

visualization.loss_visuals.create_dir(path)

Creates directories in path if it doesn’t exist. Used to store saved visualization plots.