SepiaSensitivity

The SepiaSensitivity module implements Sobol-index based sensitivity analysis.

sepia.SepiaSensitivity.sensitivity(model, samples_dict=None, ngrid=21, varlist=None, jelist=None, rg=None, option='mean')

Compute sensitivity Sobol indices. (Warning: not fully tested for all model types and input choices.)

Parameters
  • model (sepia.SepiaModel) – instantiated SepiaModel with MCMC samples

  • samples_dict (dict/NoneType) – selected samples from model.get_samples(flat=True) (default: uses all samples in model)

  • ngrid (int) – number of grid points in each dimension for calculation of mian/joint effects (default: 21)

  • varlist (list/string/NoneType) – list of tuples giving pairs of variables for which joint effects are desired; using ‘all’ indicates to compute joint effects for all variables. Default is None.

  • jelist (list) – list of tuples indicating variables for which joint sensitivities are desired (similar to varlist; default is None)

  • rg (numpy.ndarray/NoneType) – matrix with one row for each variable giving min/max values for sensitivity calculations assuming unit hypercube scaling, shape (num_vars, 2), default: unit hypercube.

  • option (string/dict) – do calculations based on ‘mean’ (posterior mean GP params), ‘median’ (posterior median GP params), ‘samples’ (GP param samples in samples_dict), or pass dict of samples from model.get_samples(flat=True)

Returns dict
Depending on input options, sens dict may contain:
  • totalMean (overall output mean – posterior mean)

  • totalVar (total output variance – posterior samples)

  • smePm (main effect sensitivity indices – posterior mean)

  • stePm (total effect sensitivity indices – posterior mean)

  • siePm (two-factor ineraction effect sensitivity indices – posterior mean)

  • sjePm (joint effect sensitivity indices – posterior mean)

  • mef (main effect functions by basis component – posterior mean and SD)

  • tmef (main effect functions – posterior mean and SD)

  • tjef (two-factor joint effect functions – posterior mean and SD)

  • sa (dict with information by basis coefficient; keys, e0 - overall output mean, vt - total output variance, sme, ste, sie, sje, mef, jef)