matsengrp / torchdms

Analyze deep mutational scanning data with PyTorch
https://matsengrp.github.io/torchdms/
2 stars 0 forks source link

Change color scheme used in heat maps of single mutant predictions #148

Closed zorian15 closed 2 years ago

zorian15 commented 2 years ago

Currently, heat maps of single mutant predictions, created via tdms heat in the CLI looks like this: Screen Shot 2021-11-30 at 3 06 34 PM

The heat maps of the beta coefficients, created via tdms beta in the CLI looks like this: Screen Shot 2021-11-30 at 3 07 30 PM

I think it would be helpful for these to match. I prefer the color scheme that Tyler used in his RBD paper -- fitness scores of 0 (or Wildtype sequences) should be white, fitness scores greater than 0 should be blue, and those less than 0 should be red.