package/gonum
1
0
Fork 0
mirror of https://github.com/gonum/gonum.git synced 2025-10-28 17:31:56 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

Update stat for the new API of MulTrans and MulVec

Browse Source
This commit is contained in:
btracey
2015-08-02 22:56:09 -06:00
parent f1b9c02aa9
commit 55a603be8c
2 changed files with 5 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

7
covariancematrix.go
View File

@@ -5,9 +5,10 @@
package stat
import (
"math"
"github.com/gonum/floats"
"github.com/gonum/matrix/mat64"
"math"
)
// CovarianceMatrix calculates a covariance matrix (also known as a
@@ -51,7 +52,7 @@ func CovarianceMatrix(cov *mat64.Dense, x mat64.Matrix, wts []float64) *mat64.De
n = float64(r)
cov.MulTrans(&xt, false, &xt, true)
cov.Mul(&xt, (&xt).T())
// Scale by the sample size.
cov.Scale(1/(n-1), cov)
@@ -74,7 +75,7 @@ func CovarianceMatrix(cov *mat64.Dense, x mat64.Matrix, wts []float64) *mat64.De
// Calculate the normalization factor.
n = floats.Sum(wts)
cov.MulTrans(&xt, false, &xt, true)
cov.Mul(&xt, (&xt).T())
// Scale by the sample size.
cov.Scale(1/(n-1), cov)