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

stat: use scaled SymOuterK

Browse Source
This commit is contained in:
kortschak
2015-09-30 16:05:17 +09:30
parent 0d0395a9f6
commit c48cebcab1
1 changed files with 6 additions and 14 deletions
Download Patch File Download Diff File Expand all files Collapse all files

20
covariancematrix.go
View File

@@ -44,15 +44,10 @@ func CovarianceMatrix(cov *mat64.SymDense, x mat64.Matrix, weights []float64) *m
floats.AddConst(-mean, v)
}
var n float64
if weights == nil {
n = float64(r)
cov.SymOuterK(&xt)
// Scale by the sample size.
cov.ScaleSym(1/(n-1), cov)
// Calculate the normalization factor
// scaled by the sample size.
cov.SymOuterK(1/(float64(r)-1), &xt)
return cov
}
@@ -70,12 +65,9 @@ func CovarianceMatrix(cov *mat64.SymDense, x mat64.Matrix, weights []float64) *m
floats.Mul(v, sqrtwts)
}
// Calculate the normalization factor.
n = floats.Sum(weights)
cov.SymOuterK(&xt)
// Scale by the sample size.
cov.ScaleSym(1/(n-1), cov)
// Calculate the normalization factor
// scaled by the weighted sample size.
cov.SymOuterK(1/(floats.Sum(weights)-1), &xt)
return cov
}