jbloomlab / polyclonal

Model mutational escape from polyclonal antibodies.
Other
3 stars 0 forks source link

customization options for escape plotting #142

Closed jbloom closed 1 year ago

jbloom commented 1 year ago

Add options to customize lineplot_and_heatmap, becomes version 3.3.

jbloom commented 1 year ago

@fwelsh @Bernadetadad, this pull request adds some options that might be useful for what you are doing.

@fwelsh, to just show certain aspects of the escape plots, you can use the new show_zoombar, show_lineplot, and show_heatmap options, which are all True by default. For instance, if you call the plotting with show_heatmap=False then you will not get a heatmap.

@Bernadetadad, there are also a few options that might make the ACE2 plotting a bit more sensible, namely by making "escape" mutations (that increase ACE2 affinity) positive (so they are easier to see / filter for) and renaming the y-axis. You could do this with the following options to the plotting: scale_stat_col=-1 (which will flip the sign on the y-axis) and `rename_stat_col="ACE2 affinity" (which will rename the y-axis).