fast_forward.interaction_plots module

Functions for plotting fitted interaction distributions

fast_forward.interaction_plots.make_distances_distribution_plot(plot_data, atom_names, save_plot_data=False, axarr=None, name='distance_distribution_plots')[source]
Parameters:

  • matrix (ndarray) – Quatratic 2D array representing the matrix

  • atom_names (list) – List of atom names corresponding to the rows and columns of the matrix

  • axarr (matplotlib.axes._axes.Axes) – array of axes to plot the fitted distributions on

  • name (str) – name of the output file (default: distribution_plots)

fast_forward.interaction_plots.make_distribution_plot(fit_data, save_plot_data=None, axarr=None, name='distribution_plots')[source]
Parameters:

  • fit_data (dict) – Dictionary containing distributions and fitting parameters for interactions. Nested as {interaction_type: {group_name: {‘data’: distribution, ‘fitted_params’: list(params)}}

  • axarr (matplotlib.axes._axes.Axes) – array of axes to plot the fitted distributions on

  • save_plot_data (bool) – if True, save the underlying data for plots as a pickle file

  • name (str) – name of the output file (default: distribution_plots)

fast_forward.interaction_plots.make_matrix_plot(matrix, atom_names, axarr=None, name='score_matrix')[source]
Parameters:

  • matrix (ndarray) – Quatratic 2D array representing the matrix

  • atom_names (list) – List of atom names corresponding to the rows and columns of the matrix

  • axarr (matplotlib.axes._axes.Axes) – array of axes to plot the fitted distributions on

  • name (str) – name of the output file (default: distribution_plots)