p3(lrls/helper.go): add addTheta,addFi funcs
This commit is contained in:
parent
a44f8def21
commit
ae69b3e306
@ -56,3 +56,50 @@ func covMatrices(size, params int) ([]*mat.Dense, error) {
|
||||
|
||||
return cms, nil
|
||||
}
|
||||
|
||||
// nolint: unused
|
||||
func newTheta(size, params int) ([][]float64, error) {
|
||||
switch {
|
||||
case size < 1:
|
||||
return nil, errors.New("ErrThetaSize")
|
||||
|
||||
case params < 4 || params > 5:
|
||||
log.Printf("error: params needs to be from the set {4 ,5}, got: %d", params)
|
||||
return nil, errors.New("ErrBadParams")
|
||||
}
|
||||
|
||||
th := make([][]float64, size)
|
||||
|
||||
// initialise.
|
||||
for i := range th {
|
||||
th[i] = make([]float64, params)
|
||||
}
|
||||
|
||||
// specifically set the second vector to all 0.1s.
|
||||
for i := range th[1] {
|
||||
th[1][i] = 0.1
|
||||
}
|
||||
|
||||
return th, nil
|
||||
}
|
||||
|
||||
// nolint: unused
|
||||
func newFi(size, params int) ([][]float64, error) {
|
||||
switch {
|
||||
case size < 1:
|
||||
return nil, errors.New("ErrFiSize")
|
||||
|
||||
case params < 4 || params > 5:
|
||||
log.Printf("error: params needs to be from the set {4 ,5}, got: %d", params)
|
||||
return nil, errors.New("ErrBadParams")
|
||||
}
|
||||
|
||||
fi := make([][]float64, size)
|
||||
|
||||
// prealloc.
|
||||
for i := range fi {
|
||||
fi[i] = make([]float64, params)
|
||||
}
|
||||
|
||||
return fi, nil
|
||||
}
|
||||
|
Loading…
Reference in New Issue
Block a user