-
-
Notifications
You must be signed in to change notification settings - Fork 1.2k
feat: add Wald distribution PDF #9324
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
Merged
Planeshifter
merged 21 commits into
stdlib-js:develop
from
manit2004:feat/wald-distribution-209
Jan 8, 2026
Merged
Changes from 14 commits
Commits
Show all changes
21 commits
Select commit
Hold shift + click to select a range
b679e63
feat: implement Wald distribution PDF #209
manit2004 3f7b53f
chore: add native.js content for Wald PDF #209
manit2004 c082919
fix: update copyright year to 2025 and add include.gypi #209
manit2004 1414bd3
chore: add manifest.json for C build configuration #209
manit2004 cc278e6
chore: add binding.gyp for native addon build #209
manit2004 e211cdb
added examples, package.json and fixed if statement flow #209
manit2004 0eeb7b7
added docs #209
manit2004 f5e2021
feat: add tests for Wald distribution PDF and fix infinity handling #209
manit2004 e368f23
feat: add tests for Wald distribution PDF with infinity handling fixes
manit2004 9cf909e
Add README and benchmarks, combine infinity checks in PDF implementation
manit2004 c049660
Refactor tests to use ULP-based tolerance and improve numerical stabi…
manit2004 90e01d1
removed unused pow func to fix linting issue
manit2004 27b3285
linting fixed in benchmark.js
manit2004 57a20cf
linting fixed in benchmark.js
manit2004 07b8345
suggestions implemented
manit2004 6b73c9e
benchmark fixed
manit2004 bf6cc88
Update copyright year to 2026
manit2004 f4dab14
Updated copyright year
manit2004 cd47482
linting fixed
manit2004 5010216
updated index.js in ./examples
manit2004 f51ec44
sugesstions implemented
manit2004 File filter
Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
There are no files selected for viewing
293 changes: 293 additions & 0 deletions
293
lib/node_modules/@stdlib/stats/base/dists/wald/pdf/README.md
This file contains hidden or bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,293 @@ | ||
| <!-- | ||
|
|
||
| @license Apache-2.0 | ||
|
|
||
| Copyright (c) 2025 The Stdlib Authors. | ||
|
|
||
| Licensed under the Apache License, Version 2.0 (the "License"); | ||
| you may not use this file except in compliance with the License. | ||
| You may obtain a copy of the License at | ||
|
|
||
| http://www.apache.org/licenses/LICENSE-2.0 | ||
|
|
||
| Unless required by applicable law or agreed to in writing, software | ||
| distributed under the License is distributed on an "AS IS" BASIS, | ||
| WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. | ||
| See the License for the specific language governing permissions and | ||
| limitations under the License. | ||
|
|
||
| --> | ||
|
|
||
| # Probability Density Function | ||
|
|
||
| > [Wald][wald-distribution] distribution probability density function (PDF). | ||
|
|
||
| <section class="intro"> | ||
|
|
||
| The [probability density function][pdf] (PDF) for a [Wald][wald-distribution] random variable is | ||
|
|
||
| <!-- <equation class="equation" label="eq:wald_pdf" align="center" raw="f(x;\mu,\lambda)=\sqrt{\frac{\lambda}{2\pi x^3}}\, e^{-\frac{\lambda(x-\mu)^2}{2\mu^2 x}}" alt="Probability density function (PDF) for a Wald distribution."> --> | ||
|
|
||
| ```math | ||
| f(x;\mu,\lambda)=\sqrt{\frac{\lambda}{2\pi x^3}}\, e^{-\frac{\lambda(x-\mu)^2}{2\mu^2 x}} | ||
| ``` | ||
|
|
||
| <!-- <div class="equation" align="center" data-raw-text="f(x;\mu,\lambda)=\sqrt{\frac{\lambda}{2\pi x^3}}\, e^{-\frac{\lambda(x-\mu)^2}{2\mu^2 x}}" data-equation="eq:wald_pdf"> | ||
| <img src="https://cdn.jsdelivr.net/gh/stdlib-js/stdlib@51534079fef45e990850102147e8945fb023d1d0/lib/node_modules/@stdlib/stats/base/dists/wald/pdf/docs/img/equation_wald_pdf.svg" alt="Probability density function (PDF) for a Wald distribution."> | ||
| <br> | ||
| </div> --> | ||
|
|
||
| <!-- </equation> --> | ||
|
|
||
| where `µ > 0` is the mean and `λ > 0` is the shape parameter. | ||
|
|
||
| </section> | ||
|
|
||
| <!-- /.intro --> | ||
|
|
||
| <section class="usage"> | ||
|
|
||
| ## Usage | ||
|
|
||
| ```javascript | ||
| var pdf = require( '@stdlib/stats/base/dists/wald/pdf' ); | ||
| ``` | ||
|
|
||
| #### pdf( x, mu, lambda ) | ||
|
|
||
| Evaluates the [probability density function][pdf] (PDF) for a [Wald][wald-distribution] distribution with parameters `mu` (mean) and `lambda` (shape parameter). | ||
|
|
||
| ```javascript | ||
| var y = pdf( 2.0, 1.0, 1.0 ); | ||
| // returns ~0.110 | ||
|
|
||
| y = pdf( 0.5, 2.0, 3.0 ); | ||
| // returns ~0.362 | ||
| ``` | ||
|
|
||
| If provided `NaN` as any argument, the function returns `NaN`. | ||
|
|
||
| ```javascript | ||
| var y = pdf( NaN, 1.0, 1.0 ); | ||
| // returns NaN | ||
|
|
||
| y = pdf( 1.0, NaN, 1.0 ); | ||
| // returns NaN | ||
|
|
||
| y = pdf( 1.0, 1.0, NaN ); | ||
| // returns NaN | ||
| ``` | ||
|
|
||
| If provided `mu <= 0`, the function returns `NaN`. | ||
|
|
||
| ```javascript | ||
| var y = pdf( 2.0, 0.0, 1.0 ); | ||
| // returns NaN | ||
|
|
||
| y = pdf( 2.0, -1.0, 1.0 ); | ||
| // returns NaN | ||
| ``` | ||
|
|
||
| If provided `lambda < 0`, the function returns `NaN`. | ||
|
|
||
| ```javascript | ||
| var y = pdf( 2.0, 1.0, -1.0 ); | ||
| // returns NaN | ||
| ``` | ||
|
|
||
| If provided `lambda = 0`, the function evaluates the [PDF][pdf] of a [degenerate distribution][degenerate-distribution] centered at `mu`. | ||
|
|
||
| ```javascript | ||
| var y = pdf( 2.0, 1.0, 0.0 ); | ||
| // returns 0.0 | ||
|
|
||
| y = pdf( 1.0, 1.0, 0.0 ); | ||
| // returns Infinity | ||
| ``` | ||
|
|
||
| If provided `x <= 0`, the function returns `0.0`. | ||
|
|
||
| ```javascript | ||
| var y = pdf( 0.0, 1.0, 1.0 ); | ||
| // returns 0.0 | ||
|
|
||
| y = pdf( -1.0, 1.0, 1.0 ); | ||
| // returns 0.0 | ||
| ``` | ||
|
|
||
| #### pdf.factory( mu, lambda ) | ||
|
|
||
| Partially applies `mu` and `lambda` to create a reusable `function` for evaluating the PDF. | ||
|
|
||
| ```javascript | ||
| var mypdf = pdf.factory( 1.0, 1.0 ); | ||
|
|
||
| var y = mypdf( 2.0 ); | ||
| // returns ~0.110 | ||
|
|
||
| y = mypdf( 0.5 ); | ||
| // returns ~0.879 | ||
| ``` | ||
|
|
||
| </section> | ||
|
|
||
| <!-- /.usage --> | ||
|
|
||
| <section class="examples"> | ||
|
|
||
| ## Examples | ||
|
|
||
| <!-- eslint no-undef: "error" --> | ||
|
|
||
| ```javascript | ||
| var randu = require( '@stdlib/random/base/randu' ); | ||
| var EPS = require( '@stdlib/constants/float64/eps' ); | ||
| var pdf = require( '@stdlib/stats/base/dists/wald/pdf' ); | ||
|
|
||
| var lambda; | ||
| var mu; | ||
| var x; | ||
| var y; | ||
| var i; | ||
|
|
||
| for ( i = 0; i < 10; i++ ) { | ||
| x = randu() * 10.0; | ||
| mu = ( randu() * 10.0 ) + EPS; | ||
| lambda = ( randu() * 20.0 ) + EPS; | ||
| y = pdf( x, mu, lambda ); | ||
| console.log( 'x: %d, µ: %d, λ: %d, f(x;µ,λ): %d', x, mu, lambda, y ); | ||
| } | ||
| ``` | ||
|
|
||
| </section> | ||
|
|
||
| <!-- /.examples --> | ||
|
|
||
| <!-- C interface documentation. --> | ||
|
|
||
| * * * | ||
|
|
||
| <section class="c"> | ||
|
|
||
| ## C APIs | ||
|
|
||
| <!-- Section to include introductory text. Make sure to keep an empty line after the intro `section` element and another before the `/section` close. --> | ||
|
|
||
| <section class="intro"> | ||
|
|
||
| </section> | ||
|
|
||
| <!-- /.intro --> | ||
|
|
||
| <!-- C usage documentation. --> | ||
|
|
||
| <section class="usage"> | ||
|
|
||
| ### Usage | ||
|
|
||
| ```c | ||
| #include "stdlib/stats/base/dists/wald/pdf.h" | ||
| ``` | ||
|
|
||
| #### stdlib_base_dists_wald_pdf( x, mu, lambda ) | ||
|
|
||
| Evaluates the [probability density function][pdf] (PDF) for a [Wald][wald-distribution] distribution with parameters `mu` (mean) and `lambda` (shape parameter). | ||
|
|
||
| ```c | ||
| double y = stdlib_base_dists_wald_pdf( 2.0, 1.0, 1.0 ); | ||
| // returns ~0.110 | ||
| ``` | ||
|
|
||
| The function accepts the following arguments: | ||
|
|
||
| - **x**: `[in] double` input value. | ||
| - **mu**: `[in] double` mean. | ||
| - **lambda**: `[in] double` shape parameter. | ||
|
|
||
| ```c | ||
| double stdlib_base_dists_wald_pdf( const double x, const double mu, const double lambda ); | ||
| ``` | ||
|
|
||
| </section> | ||
|
|
||
| <!-- /.usage --> | ||
|
|
||
| <!-- C API usage notes. Make sure to keep an empty line after the `section` element and another before the `/section` close. --> | ||
|
|
||
| <section class="notes"> | ||
|
|
||
| </section> | ||
|
|
||
| <!-- /.notes --> | ||
|
|
||
| <!-- C API usage examples. --> | ||
|
|
||
| <section class="examples"> | ||
|
|
||
| ### Examples | ||
|
|
||
| ```c | ||
| #include "stdlib/stats/base/dists/wald/pdf.h" | ||
| #include "stdlib/constants/float64/eps.h" | ||
| #include <stdlib.h> | ||
| #include <stdio.h> | ||
|
|
||
| static double random_uniform( const double min, const double max ) { | ||
| double v = (double)rand() / ( (double)RAND_MAX + 1.0 ); | ||
| return min + ( v*(max-min) ); | ||
| } | ||
|
|
||
| int main( void ) { | ||
| double lambda; | ||
| double mu; | ||
| double x; | ||
| double y; | ||
| int i; | ||
|
|
||
| for ( i = 0; i < 10; i++ ) { | ||
| x = random_uniform( 0.0, 10.0 ); | ||
| mu = random_uniform( STDLIB_CONSTANT_FLOAT64_EPS, 10.0 ); | ||
| lambda = random_uniform( STDLIB_CONSTANT_FLOAT64_EPS, 20.0 ); | ||
| y = stdlib_base_dists_wald_pdf( x, mu, lambda ); | ||
| printf( "x: %lf, µ: %lf, λ: %lf, f(x;µ,λ): %lf\n", x, mu, lambda, y ); | ||
| } | ||
| } | ||
| ``` | ||
|
|
||
| </section> | ||
|
|
||
| <!-- /.examples --> | ||
|
|
||
| </section> | ||
|
|
||
| <!-- /.c --> | ||
|
|
||
| <!-- Section to include cited references. If references are included, add a horizontal rule *before* the section. Make sure to keep an empty line after the `section` element and another before the `/section` close. --> | ||
|
|
||
| <section class="references"> | ||
|
|
||
| </section> | ||
|
|
||
| <!-- /.references --> | ||
|
|
||
| <!-- Section for related `stdlib` packages. Do not manually edit this section, as it is automatically populated. --> | ||
|
|
||
| <section class="related"> | ||
|
|
||
| </section> | ||
|
|
||
| <!-- /.related --> | ||
|
|
||
| <!-- Section for all links. Make sure to keep an empty line after the `section` element and another before the `/section` close. --> | ||
|
|
||
| <section class="links"> | ||
|
|
||
| [pdf]: https://en.wikipedia.org/wiki/Probability_density_function | ||
|
|
||
| [wald-distribution]: https://en.wikipedia.org/wiki/Inverse_Gaussian_distribution | ||
|
|
||
| [degenerate-distribution]: https://en.wikipedia.org/wiki/Degenerate_distribution | ||
|
|
||
| </section> | ||
|
|
||
| <!-- /.links --> | ||
93 changes: 93 additions & 0 deletions
93
lib/node_modules/@stdlib/stats/base/dists/wald/pdf/benchmark/benchmark.js
This file contains hidden or bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,93 @@ | ||
| /** | ||
| * @license Apache-2.0 | ||
| * | ||
| * Copyright (c) 2025 The Stdlib Authors. | ||
| * | ||
| * Licensed under the Apache License, Version 2.0 (the "License"); | ||
| * you may not use this file except in compliance with the License. | ||
| * You may obtain a copy of the License at | ||
| * | ||
| * http://www.apache.org/licenses/LICENSE-2.0 | ||
| * | ||
| * Unless required by applicable law or agreed to in writing, software | ||
| * distributed under the License is distributed on an "AS IS" BASIS, | ||
| * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. | ||
| * See the License for the specific language governing permissions and | ||
| * limitations under the License. | ||
| */ | ||
|
|
||
| 'use strict'; | ||
|
|
||
| // MODULES // | ||
|
|
||
| var bench = require( '@stdlib/bench' ); | ||
| var Float64Array = require( '@stdlib/array/float64' ); | ||
| var uniform = require( '@stdlib/random/base/uniform' ); | ||
| var isnan = require( '@stdlib/math/base/assert/is-nan' ); | ||
| var EPS = require( '@stdlib/constants/float64/eps' ); | ||
| var pkg = require( './../package.json' ).name; | ||
| var pdf = require( './../lib' ); | ||
|
|
||
|
|
||
| // MAIN // | ||
|
|
||
| bench( pkg, function benchmark( b ) { | ||
| var lambda; | ||
| var len; | ||
| var mu; | ||
| var x; | ||
| var y; | ||
| var i; | ||
|
|
||
| len = 100; | ||
| x = new Float64Array( len ); | ||
|
Member
There was a problem hiding this comment. Choose a reason for hiding this commentThe reason will be displayed to describe this comment to others. Learn more. See other packages for potential ways to refactor this. |
||
| mu = new Float64Array( len ); | ||
| lambda = new Float64Array( len ); | ||
| for ( i = 0; i < len; i++ ) { | ||
| x[ i ] = uniform( EPS, 100.0 ); | ||
| mu[ i ] = uniform( EPS, 50.0 ); | ||
| lambda[ i ] = uniform( EPS, 20.0 ); | ||
| } | ||
|
|
||
| b.tic(); | ||
| for ( i = 0; i < b.iterations; i++ ) { | ||
| y = pdf( x[ i%len ], mu[ i%len ], lambda[ i%len ] ); | ||
| if ( isnan( y ) ) { | ||
| b.fail( 'should not return NaN' ); | ||
| } | ||
| } | ||
| b.toc(); | ||
| if ( isnan( y ) ) { | ||
| b.fail( 'should not return NaN' ); | ||
| } | ||
| b.pass( 'benchmark finished' ); | ||
| b.end(); | ||
| }); | ||
|
|
||
| bench( pkg+':factory', function benchmark( b ) { | ||
|
Member
There was a problem hiding this comment. Choose a reason for hiding this commentThe reason will be displayed to describe this comment to others. Learn more. Use |
||
| var lambda; | ||
| var mypdf; | ||
| var mu; | ||
| var x; | ||
| var y; | ||
| var i; | ||
|
|
||
| mu = 1.0; | ||
| lambda = 1.5; | ||
| mypdf = pdf.factory( mu, lambda ); | ||
|
|
||
| b.tic(); | ||
| for ( i = 0; i < b.iterations; i++ ) { | ||
| x = uniform( EPS, 10.0 ); | ||
| y = mypdf( x ); | ||
| if ( isnan( y ) ) { | ||
| b.fail( 'should not return NaN' ); | ||
| } | ||
| } | ||
| b.toc(); | ||
| if ( isnan( y ) ) { | ||
| b.fail( 'should not return NaN' ); | ||
| } | ||
| b.pass( 'benchmark finished' ); | ||
| b.end(); | ||
| }); | ||
Oops, something went wrong.
Oops, something went wrong.
Add this suggestion to a batch that can be applied as a single commit.
This suggestion is invalid because no changes were made to the code.
Suggestions cannot be applied while the pull request is closed.
Suggestions cannot be applied while viewing a subset of changes.
Only one suggestion per line can be applied in a batch.
Add this suggestion to a batch that can be applied as a single commit.
Applying suggestions on deleted lines is not supported.
You must change the existing code in this line in order to create a valid suggestion.
Outdated suggestions cannot be applied.
This suggestion has been applied or marked resolved.
Suggestions cannot be applied from pending reviews.
Suggestions cannot be applied on multi-line comments.
Suggestions cannot be applied while the pull request is queued to merge.
Suggestion cannot be applied right now. Please check back later.
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.
See #9439 for an example of how to replace manual
forloops.