torchdms.plot.beta_coefficients

torchdms.plot.beta_coefficients(model, test_data, out)

This function takes in a (ideally trained) model and plots the values of the weights corresponding to the inputs, dubbed “beta coefficients”.

We plot this as a heatmap where the rows are substitutions nucleotides, and the columns are the sequence positions for each mutation.