@stdlib/stats-base-dists-negative-binomial-pmf
TypeScript icon, indicating that this package has built-in type declarations

0.1.1 • Public • Published
About stdlib...

We believe in a future in which the web is a preferred environment for numerical computation. To help realize this future, we've built stdlib. stdlib is a standard library, with an emphasis on numerical and scientific computation, written in JavaScript (and C) for execution in browsers and in Node.js.

The library is fully decomposable, being architected in such a way that you can swap out and mix and match APIs and functionality to cater to your exact preferences and use cases.

When you use stdlib, you can be absolutely certain that you are using the most thorough, rigorous, well-written, studied, documented, tested, measured, and high-quality code out there.

To join us in bringing numerical computing to the web, get started by checking us out on GitHub, and please consider financially supporting stdlib. We greatly appreciate your continued support!

Probability Mass Function

NPM version Build Status Coverage Status

Negative binomial distribution probability mass function (PMF).

The probability mass function (PMF) for a negative binomial random variable X is

Probability mass function (PMF) for a negative binomial distribution.

where r > 0 is the number of successes until experiment is stopped and 0 < p <= 1 is the success probability. The random variable X denotes the number of failures until the r success is reached.

Installation

npm install @stdlib/stats-base-dists-negative-binomial-pmf

Usage

var pmf = require( '@stdlib/stats-base-dists-negative-binomial-pmf' );

pmf( x, r, p )

Evaluates the probability mass function for a negative binomial distribution with number of successes until experiment is stopped r and success probability p.

var y = pmf( 5.0, 20.0, 0.8 );
// returns ~0.157

y = pmf( 21.0, 20.0, 0.5 );
// returns ~0.06

y = pmf( 5.0, 10.0, 0.4 );
// returns ~0.016

y = pmf( 0.0, 10.0, 0.9 );
// returns ~0.349

While r can be interpreted as the number of successes until the experiment is stopped, the negative binomial distribution is also defined for non-integers r. In this case, r denotes shape parameter of the gamma mixing distribution.

var y = pmf( 21.0, 15.5, 0.5 );
// returns ~0.037

y = pmf( 5.0, 7.4, 0.4 );
// returns ~0.051

If provided a r which is not a positive number, the function returns NaN.

var y = pmf( 2.0, 0.0, 0.5 );
// returns NaN

y = pmf( 2.0, -2.0, 0.5 );
// returns NaN

If provided NaN as any argument, the function returns NaN.

var y = pmf( NaN, 20.0, 0.5 );
// returns NaN

y = pmf( 0.0, NaN, 0.5 );
// returns NaN

y = pmf( 0.0, 20.0, NaN );
// returns NaN

If provided a success probability p outside of [0,1], the function returns NaN.

var y = pmf( 2.0, 20, -1.0 );
// returns NaN

y = pmf( 2.0, 20, 1.5 );
// returns NaN

pmf.factory( r, p )

Returns a function for evaluating the probability mass function (PMF) of a negative binomial distribution with number of successes until experiment is stopped r and success probability p.

var mypmf = pmf.factory( 10, 0.5 );
var y = mypmf( 3.0 );
// returns ~0.03

y = mypmf( 10.0 );
// returns ~0.088

Examples

var randu = require( '@stdlib/random-base-randu' );
var round = require( '@stdlib/math-base-special-round' );
var pmf = require( '@stdlib/stats-base-dists-negative-binomial-pmf' );

var i;
var r;
var p;
var x;
var y;

for ( i = 0; i < 10; i++ ) {
    x = round( randu() * 30 );
    r = randu() * 50;
    p = randu();
    y = pmf( x, r, p );
    console.log( 'x: %d, r: %d, p: %d, P(X=x;r,p): %d', x, r, p.toFixed( 4 ), y.toFixed( 4 ) );
}

Notice

This package is part of stdlib, a standard library for JavaScript and Node.js, with an emphasis on numerical and scientific computing. The library provides a collection of robust, high performance libraries for mathematics, statistics, streams, utilities, and more.

For more information on the project, filing bug reports and feature requests, and guidance on how to develop stdlib, see the main project repository.

Community

Chat


Copyright

Copyright © 2016-2024. The Stdlib Authors.

Package Sidebar

Install

npm i @stdlib/stats-base-dists-negative-binomial-pmf

Homepage

stdlib.io

Weekly Downloads

136

Version

0.1.1

License

Apache-2.0

Unpacked Size

79.9 kB

Total Files

15

Last publish

Collaborators

  • stdlib-bot
  • kgryte
  • planeshifter
  • rreusser