torchdms.plot.plot_heatmap

torchdms.plot.plot_heatmap(model, path)

This function takes in a model and plots the single mutant predictions.

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