rail.estimation.algos.varInference module

A summarizer that simple makes a histogram of a point estimate

class rail.estimation.algos.varInference.VarInferenceStack(args, comm=None)[source]

Bases: PZSummarizer

Variational inference summarizer based on notebook created by Markus Rau The summzarizer is appropriate for the likelihoods returned by template-based codes, for which the NaiveSummarizer are not appropriate.

Parameters:

  • zmin (float) – minimum z for redshift grid

  • zmax (float) – maximum z for redshift grid

  • nzbins (int) – number of bins for redshift grid

  • niter (int) – number of iterations to perform in the variational inference

  • nsamples (int) – number of samples used in dirichlet to determind error bar

config_options = {'chunk_size': 10000, 'niter': <ceci.config.StageParameter object>, 'nsamples': <ceci.config.StageParameter object>, 'nzbins': <ceci.config.StageParameter object>, 'output_mode': <ceci.config.StageParameter object>, 'seed': <ceci.config.StageParameter object>, 'zmax': <ceci.config.StageParameter object>, 'zmin': <ceci.config.StageParameter object>}
name = 'VarInferenceStack'
outputs = [('output', <class 'rail.core.data.QPHandle'>), ('single_NZ', <class 'rail.core.data.QPHandle'>)]
run()[source]

Run the stage and return the execution status