InfectionMedicineProteomics / BINN

Generating biologically informed neural networks
https://infectionmedicineproteomics.github.io/BINN/
MIT License
25 stars 10 forks source link

Minor changes to plotting #20

Closed ErikHartman closed 1 year ago

ErikHartman commented 1 year ago
  1. Add image dimensions as arguments.
  2. Add numeric_only=True to pandas groupby. This fixes compatability with pandas 2.0
  3. If multiclass == False, we sum SHAP values. This makes it so that there is only 1 flow if the source and target is the same.
  4. Merged if several edges are coming and going to the same nodes. This is relevant for edges incoming to "Other connections" since these are lumped to a single edge.