export_train_plots

behavenet.fitting.eval.export_train_plots(hparams, dtype, loss_type='mse', save_file=None, format='png')

Export plot with MSE/LL as a function of training epochs.

Parameters:

• hparams (dict) – needs to contain enough information to specify the desired model (autoencoder, arhmm, etc.)

• dtype (str) – type of trials to use for plotting: ‘train’ | ‘val’ (metrics are not computed for ‘test’ trials throughout training)

• loss_type (str, optional) – ‘mse’ | ‘ll’

• save_file (str or NoneType, optional) – full filename (absolute path) for saving plot; if NoneType, plot is displayed

• format (str) – file format of plot, e.g. ‘png’ | ‘pdf’ | ‘jpeg’