InfectionMedicineProteomics / BINN

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

Added explain_average to BINNExplainer #3

Closed ErikHartman closed 1 year ago

ErikHartman commented 1 year ago

This trains the network N times (reseting parameters and initializing after each iteration) and computes SHAP values for all trained models. It then aggregates this into one dataframe with mean and std. The 'value' columns is identical to values_mean and used for plotting