-
-
Notifications
You must be signed in to change notification settings - Fork 2k
New issue
Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.
By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.
Already on GitHub? Sign in to your account
Drop nuts init method from pm.sample #3863
Conversation
Codecov Report
@@ Coverage Diff @@
## master #3863 +/- ##
==========================================
- Coverage 90.73% 90.69% -0.04%
==========================================
Files 135 135
Lines 21190 21185 -5
==========================================
- Hits 19227 19214 -13
- Misses 1963 1971 +8
|
@@ -1865,14 +1864,12 @@ def init_nuts( | |||
* advi: Run ADVI to estimate posterior mean and diagonal mass matrix. | |||
* advi_map: Initialize ADVI with MAP and use MAP as starting point. | |||
* map: Use the MAP as starting point. This is discouraged. | |||
* nuts: Run NUTS and estimate posterior mean and mass matrix from | |||
the trace. | |||
* adapt_full: Adapt a dense mass matrix using the sample covariances | |||
chains: int | |||
Number of jobs to start. | |||
n_init: int |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
I think n_init
is only used for advi now, right? COuld make that clear in the doc string.
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Done
Thanks! Also release-notes :). |
Thanks! |
* Added wrapper for gamma sampling in negative binomial distribution. * Fixed typo in posterior sampling warnings. * Drop nuts init method from pm.sample (#3863) * Dropped nuts init method * Dropped nuts init method from tests * Refined doc string and added release note * Added wrapper for gamma sampling in negative binomial distribution. * Fixed typo in posterior sampling warnings. * Added test for negative binomial input shape. * Updating release notes with changes to negative binomial sampling. Co-authored-by: Alexandre ANDORRA <[email protected]>
As discussed in #3861, the nuts init method is a left-over in
pm.sample
init methods and should just be dropped. This PR does it.Thanks in advance for the review, and tell me if any edit is needed 🖖