Chi-squared distribution moment-generating function (MGF).
The moment-generating function for a Chi-squared random variable is
where k
is the degrees of freedom.
$ npm install distributions-chisquare-mgf
For use in the browser, use browserify.
var mgf = require( 'distributions-chisquare-mgf' );
Evaluates the moment-generating function (MGF) for the Chi-squared distribution. t
may be either a number
, an array
, a typed array
, or a matrix
.
var matrix = require( 'dstructs-matrix' ),
mat,
out,
t,
i;
out = mgf( 0.2 );
// returns ~1.291
out = mgf( -1 );
// returns ~0.577
t = [ 0, 0.2, 0.4, 0.6, 0.8, 1 ];
out = mgf( t );
// returns [ 1, ~1.291, ~2.236, NaN, NaN, NaN ]
t = new Float32Array( t );
out = mgf( t );
// returns Float64Array( [1,~1.291,~2.236,NaN,NaN,NaN] )
t = new Float32Array( 6 );
for ( i = 0; i < 6; i++ ) {
t[ i ] = i * 0.2;
}
mat = matrix( t, [3,2], 'float32' );
/*
[ 0 0.2
0.4 0.6
0.8 1 ]
*/
out = mgf( mat );
/*
[ 1 ~1.291
~2.236 NaN
NaN NaN ]
*/
The function accepts the following options
:
- k: degrees of freedom. Default:
1
. - accessor: accessor
function
for accessingarray
values. - dtype: output
typed array
ormatrix
data type. Default:float64
. - copy:
boolean
indicating if thefunction
should return a new data structure. Default:true
. - path: deepget/deepset key path.
- sep: deepget/deepset key path separator. Default:
'.'
.
A Chi-squared distribution is a function of 1 parameter(s): k
(degrees of freedom). By default, k
is equal to 1
. To adjust either parameter, set the corresponding option.
var t = [ 0, 0.2, 0.4, 0.6, 0.8, 1 ];
var out = mgf( t, {
'k': 3
});
// returns [ 1, ~2.152, ~11.18, NaN, NaN, NaN ]
For non-numeric arrays
, provide an accessor function
for accessing array
values.
var data = [
[0,0],
[1,0.2],
[2,0.4],
[3,0.6],
[4,0.8],
[5,1]
];
function getValue( d, i ) {
return d[ 1 ];
}
var out = mgf( data, {
'accessor': getValue
});
// returns [ 1, ~1.291, ~2.236, NaN, NaN, NaN ]
To deepset an object array
, provide a key path and, optionally, a key path separator.
var data = [
{'x':[0,0]},
{'x':[1,0.2]},
{'x':[2,0.4]},
{'x':[3,0.6]},
{'x':[4,0.8]},
{'x':[5,1]}
];
var out = mgf( data, {
'path': 'x/1',
'sep': '/'
});
/*
[
{'x':[0,1]},
{'x':[1,~1.291]},
{'x':[2,~2.236]},
{'x':[3,NaN]},
{'x':[4,NaN]},
{'x':[5,NaN]}
]
*/
var bool = ( data === out );
// returns true
By default, when provided a typed array
or matrix
, the output data structure is float64
in order to preserve precision. To specify a different data type, set the dtype
option (see matrix
for a list of acceptable data types).
var t, out;
t = new Int8Array( [0,0.2,0.4] );
out = mgf( t, {
'dtype': 'int32'
});
// returns Int32Array( [1,1,2] )
// Works for plain arrays, as well...
out = mgf( [0,0.2,0.4], {
'dtype': 'uint8'
});
// returns Uint8Array( [1,1,2] )
By default, the function returns a new data structure. To mutate the input data structure (e.g., when input values can be discarded or when optimizing memory usage), set the copy
option to false
.
var bool,
mat,
out,
t,
i;
t = [ 0, 0.2, 0.4, 0.6, 0.8, 1 ];
out = mgf( t, {
'copy': false
});
// returns [ 1, ~1.291, ~2.236, NaN, NaN, NaN ]
bool = ( t === out );
// returns true
t = new Float32Array( 6 );
for ( i = 0; i < 6; i++ ) {
t[ i ] = i * 0.2;
}
mat = matrix( t, [3,2], 'float32' );
/*
[ 0 0.2
0.4 0.6
0.8 1 ]
*/
out = mgf( mat, {
'copy': false
});
/*
[ 1 ~1.291
~2.236 NaN
NaN NaN ]
*/
bool = ( mat === out );
// returns true
-
If an element is not a numeric value, the evaluated MGF is
NaN
.var data, out; out = mgf( null ); // returns NaN out = mgf( true ); // returns NaN out = mgf( {'a':'b'} ); // returns NaN out = mgf( [ true, null, [] ] ); // returns [ NaN, NaN, NaN ] function getValue( d, i ) { return d.x; } data = [ {'x':true}, {'x':[]}, {'x':{}}, {'x':null} ]; out = mgf( data, { 'accessor': getValue }); // returns [ NaN, NaN, NaN, NaN ] out = mgf( data, { 'path': 'x' }); /* [ {'x':NaN}, {'x':NaN}, {'x':NaN, {'x':NaN} ] */
-
Be careful when providing a data structure which contains non-numeric elements and specifying an
integer
output data type, asNaN
values are cast to0
.var out = mgf( [ true, null, [] ], { 'dtype': 'int8' }); // returns Int8Array( [0,0,0] );
var mgf = require( 'distributions-chisquare-mgf' ),
matrix = require( 'dstructs-matrix' );
var data,
mat,
out,
tmp,
i;
// Plain arrays...
data = new Array( 10 );
for ( i = 0; i < data.length; i++ ) {
data[ i ] = i / 20;
}
out = mgf( data );
// Object arrays (accessors)...
function getValue( d ) {
return d.x;
}
for ( i = 0; i < data.length; i++ ) {
data[ i ] = {
'x': data[ i ]
};
}
out = mgf( data, {
'accessor': getValue
});
// Deep set arrays...
for ( i = 0; i < data.length; i++ ) {
data[ i ] = {
'x': [ i, data[ i ].x ]
};
}
out = mgf( data, {
'path': 'x/1',
'sep': '/'
});
// Typed arrays...
data = new Float32Array( 10 );
for ( i = 0; i < data.length; i++ ) {
data[ i ] = i / 20;
}
out = mgf( data );
// Matrices...
mat = matrix( data, [5,2], 'float32' );
out = mgf( mat );
// Matrices (custom output data type)...
out = mgf( mat, {
'dtype': 'uint8'
});
To run the example code from the top-level application directory,
$ node ./examples/index.js
Unit tests use the Mocha test framework with Chai assertions. To run the tests, execute the following command in the top-level application directory:
$ make test
All new feature development should have corresponding unit tests to validate correct functionality.
This repository uses Istanbul as its code coverage tool. To generate a test coverage report, execute the following command in the top-level application directory:
$ make test-cov
Istanbul creates a ./reports/coverage
directory. To access an HTML version of the report,
$ make view-cov
Copyright © 2015. The Compute.io Authors.