Skip to content

Commit

Permalink
Updating default inputs for estimators. (#59)
Browse files Browse the repository at this point in the history
* Updaing default inputs for estimators.

* Also overriding `inputs` for var_inf.
  • Loading branch information
drewoldag authored Oct 25, 2023
1 parent 069e09d commit 88f1385
Show file tree
Hide file tree
Showing 3 changed files with 3 additions and 0 deletions.
1 change: 1 addition & 0 deletions src/rail/estimation/algos/naive_stack.py
Original file line number Diff line number Diff line change
Expand Up @@ -35,6 +35,7 @@ class NaiveStackSummarizer(PZSummarizer):
nzbins=Param(int, 301, msg="The number of gridpoints in the z grid"),
seed=Param(int, 87, msg="random seed"),
nsamples=Param(int, 1000, msg="Number of sample distributions to create"))
inputs = [('input', QPHandle)]
outputs = [('output', QPHandle),
('single_NZ', QPHandle)]

Expand Down
1 change: 1 addition & 0 deletions src/rail/estimation/algos/point_est_hist.py
Original file line number Diff line number Diff line change
Expand Up @@ -37,6 +37,7 @@ class PointEstHistSummarizer(PZSummarizer):
seed=Param(int, 87, msg="random seed"),
point_estimate=Param(str, 'zmode', msg="Which point estimate to use"),
nsamples=Param(int, 1000, msg="Number of sample distributions to return"))
input = [('input', QPHandle)]
outputs = [('output', QPHandle),
('single_NZ', QPHandle)]

Expand Down
1 change: 1 addition & 0 deletions src/rail/estimation/algos/var_inf.py
Original file line number Diff line number Diff line change
Expand Up @@ -55,6 +55,7 @@ class VarInfStackSummarizer(PZSummarizer):
seed=Param(int, 87, msg="random seed"),
niter=Param(int, 100, msg="The number of iterations in the variational inference"),
nsamples=Param(int, 500, msg="The number of samples used in dirichlet uncertainty"))
inputs = [('input', QPHandle)]
outputs = [('output', QPHandle),
('single_NZ', QPHandle)]

Expand Down

0 comments on commit 88f1385

Please sign in to comment.