コード例 #1
0
func TestBaseForwardInvert(t *testing.T) {
	uncertainty := &base{
		tasks: []uint{0, 1, 2},
		lower: []float64{10.0, 20.0, 30.0},
		upper: []float64{20.0, 30.0, 40.0},

		nt: 3,
		nu: 3,
		nz: 2,

		copula: &copula{
			C: []float64{
				1.0, 2.0, 3.0,
				4.0, 5.0, 6.0,
			},
			D: []float64{
				6.0, 5.0,
				4.0, 3.0,
				2.0, 1.0,
			},
		},
		marginals: []distribution.Continuous{
			distribution.NewUniform(0.0, 1.0),
			distribution.NewUniform(0.0, 1.0),
			distribution.NewUniform(0.0, 1.0),
		},
	}

	forward := uncertainty.Forward([]float64{18.0, 21.0, 36.0})
	assert.EqualWithin(forward, []float64{
		6.664804998759882e-01,
		7.313162037785672e-01,
	}, 1e-14, t)

	backward := uncertainty.Backward([]float64{0.45, 0.65})
	assert.EqualWithin(backward, []float64{
		1.921556679782504e+01,
		2.953060310728164e+01,
		3.973501094321997e+01,
	}, 1e-14, t)
}
コード例 #2
0
ファイル: main.go プロジェクト: turing-complete/laboratory
func Parse(line string) (distribution.Continuous, error) {
	family, params := parse(line)

	switch family {
	case betaFamily:
		return distribution.NewBeta(params[0], params[1], 0.0, 1.0), nil
	case uniformFamily:
		return distribution.NewUniform(0.0, 1.0), nil
	default:
		return nil, errors.New("the marginal distribution is unknown")
	}
}