34 lines
1.0 KiB
ReasonML
34 lines
1.0 KiB
ReasonML
type assumption =
|
|
| ADDS_TO_1
|
|
| ADDS_TO_CORRECT_PROBABILITY;
|
|
|
|
type assumptions = {
|
|
continuous: assumption,
|
|
discrete: assumption,
|
|
discreteProbabilityMass: option(float),
|
|
};
|
|
|
|
let buildSimple = (~continuous: option(DistTypes.continuousShape), ~discrete: option(DistTypes.discreteShape)): option(DistTypes.shape) => {
|
|
let continuous = continuous |> E.O.default(Continuous.make(~integralSumCache=Some(0.0), {xs: [||], ys: [||]}));
|
|
let discrete = discrete |> E.O.default(Discrete.make(~integralSumCache=Some(0.0), {xs: [||], ys: [||]}));
|
|
let cLength =
|
|
continuous
|
|
|> Continuous.getShape
|
|
|> XYShape.T.xs
|
|
|> E.A.length;
|
|
let dLength = discrete |> Discrete.getShape |> XYShape.T.xs |> E.A.length;
|
|
switch (cLength, dLength) {
|
|
| (0 | 1, 0) => None
|
|
| (0 | 1, _) => Some(Discrete(discrete))
|
|
| (_, 0) => Some(Continuous(continuous))
|
|
| (_, _) =>
|
|
let mixedDist =
|
|
Mixed.make(
|
|
~integralSumCache=None,
|
|
~integralCache=None,
|
|
~continuous,
|
|
~discrete,
|
|
);
|
|
Some(Mixed(mixedDist));
|
|
};
|
|
}; |