Fix unweighted average of distributions
This commit is contained in:
parent
38135f0c81
commit
117c08bfa9
|
@ -3,7 +3,8 @@
|
||||||
"name": "squiggle",
|
"name": "squiggle",
|
||||||
"scripts": {
|
"scripts": {
|
||||||
"nodeclean": "rm -r node_modules && rm -r packages/*/node_modules",
|
"nodeclean": "rm -r node_modules && rm -r packages/*/node_modules",
|
||||||
"format:all": "prettier --write . && cd packages/squiggle-lang && yarn format"
|
"format:all": "prettier --write . && cd packages/squiggle-lang && yarn format",
|
||||||
|
"lint:all": "prettier --check . && cd packages/squiggle-lang && yarn lint:rescript"
|
||||||
},
|
},
|
||||||
"devDependencies": {
|
"devDependencies": {
|
||||||
"prettier": "^2.6.2"
|
"prettier": "^2.6.2"
|
||||||
|
|
4
packages/squiggle-lang/.prettierignore
Normal file
4
packages/squiggle-lang/.prettierignore
Normal file
|
@ -0,0 +1,4 @@
|
||||||
|
dist
|
||||||
|
lib
|
||||||
|
*.bs.js
|
||||||
|
*.gen.tsx
|
|
@ -90,16 +90,8 @@ describe("eval on distribution functions", () => {
|
||||||
})
|
})
|
||||||
|
|
||||||
describe("mixture", () => {
|
describe("mixture", () => {
|
||||||
testEval(
|
testEval("mx(normal(5,2), normal(10,1), normal(15, 1))", "Ok(Point Set Distribution)")
|
||||||
~skip=true,
|
testEval("mixture(normal(5,2), normal(10,1), [0.2, 0.4])", "Ok(Point Set Distribution)")
|
||||||
"mx(normal(5,2), normal(10,1), normal(15, 1))",
|
|
||||||
"Ok(Point Set Distribution)",
|
|
||||||
)
|
|
||||||
testEval(
|
|
||||||
~skip=true,
|
|
||||||
"mixture(normal(5,2), normal(10,1), [.2,, .4])",
|
|
||||||
"Ok(Point Set Distribution)",
|
|
||||||
)
|
|
||||||
})
|
})
|
||||||
})
|
})
|
||||||
|
|
||||||
|
|
|
@ -66,36 +66,62 @@ module Helpers = {
|
||||||
dist1,
|
dist1,
|
||||||
)->runGenericOperation
|
)->runGenericOperation
|
||||||
}
|
}
|
||||||
let parseNumber = (args: expressionValue) : Belt.Result.t<float, string> =>
|
let parseNumber = (args: expressionValue): Belt.Result.t<float, string> =>
|
||||||
switch args {
|
switch args {
|
||||||
| EvNumber(x) => Ok(x)
|
| EvNumber(x) => Ok(x)
|
||||||
| _ => Error("Not a number")
|
| _ => Error("Not a number")
|
||||||
}
|
}
|
||||||
|
|
||||||
let parseNumberArray = (ags: array<expressionValue>) : Belt.Result.t<array<float>, string> => E.A.fmap(parseNumber, ags) |> E.A.R.firstErrorOrOpen
|
let parseNumberArray = (ags: array<expressionValue>): Belt.Result.t<array<float>, string> =>
|
||||||
|
E.A.fmap(parseNumber, ags) |> E.A.R.firstErrorOrOpen
|
||||||
|
|
||||||
let parseDist = (args: expressionValue): Belt.Result.t<GenericDist_Types.genericDist, string> =>
|
let parseDist = (args: expressionValue): Belt.Result.t<GenericDist_Types.genericDist, string> =>
|
||||||
switch args {
|
switch args {
|
||||||
| EvDistribution(x) => Ok(x)
|
| EvDistribution(x) => Ok(x)
|
||||||
| EvNumber(x) => Ok(GenericDist.fromFloat(x))
|
| EvNumber(x) => Ok(GenericDist.fromFloat(x))
|
||||||
| _ => Error("Not a distribution")
|
| _ => Error("Not a distribution")
|
||||||
}
|
}
|
||||||
|
|
||||||
let parseDistributionArray = (ags: array<expressionValue>) : Belt.Result.t<array<GenericDist_Types.genericDist>, string> => E.A.fmap(parseDist, ags) |> E.A.R.firstErrorOrOpen
|
let parseDistributionArray = (ags: array<expressionValue>): Belt.Result.t<
|
||||||
|
array<GenericDist_Types.genericDist>,
|
||||||
|
string,
|
||||||
|
> => E.A.fmap(parseDist, ags) |> E.A.R.firstErrorOrOpen
|
||||||
|
|
||||||
let mixture = (args : array<expressionValue>): DistributionOperation.outputType => {
|
let mixtureWithGivenWeights = (
|
||||||
let givenWeights = E.A.last(args)
|
distributions: array<GenericDist_Types.genericDist>,
|
||||||
let calculatedWeights =
|
weights: array<float>,
|
||||||
switch givenWeights {
|
): DistributionOperation.outputType =>
|
||||||
| Some(EvArray(b)) => parseNumberArray(b)
|
E.A.length(distributions) == E.A.length(weights)
|
||||||
| None =>
|
? Mixture(Belt.Array.zip(distributions, weights))->runGenericOperation
|
||||||
Ok(Belt.Array.make(E.A.length(args), 1.0 /. Belt.Int.toFloat(E.A.length(args))))
|
: GenDistError(
|
||||||
| _ => Error("Last argument of mx must be array")
|
ArgumentError("Error, mixture call has different number of distributions and weights"),
|
||||||
|
)
|
||||||
|
|
||||||
|
let mixtureWithDefaultWeights = (
|
||||||
|
distributions: array<GenericDist_Types.genericDist>,
|
||||||
|
): DistributionOperation.outputType => {
|
||||||
|
let length = E.A.length(distributions)
|
||||||
|
let weights = Belt.Array.make(length, 1.0 /. Belt.Int.toFloat(length))
|
||||||
|
mixtureWithGivenWeights(distributions, weights)
|
||||||
|
}
|
||||||
|
|
||||||
|
let mixture = (args: array<expressionValue>): DistributionOperation.outputType => {
|
||||||
|
switch E.A.last(args) {
|
||||||
|
| Some(EvArray(b)) => {
|
||||||
|
let weights = parseNumberArray(b)
|
||||||
|
let distributions = parseDistributionArray(
|
||||||
|
Belt.Array.slice(args, ~offset=0, ~len=E.A.length(args) - 1),
|
||||||
|
)
|
||||||
|
switch E.R.merge(distributions, weights) {
|
||||||
|
| Ok(d, w) => mixtureWithGivenWeights(d, w)
|
||||||
|
| Error(err) => GenDistError(ArgumentError(err))
|
||||||
}
|
}
|
||||||
switch (parseDistributionArray(Belt.Array.slice(args, ~offset=0, ~len=Belt.Array.length(args)-1)), calculatedWeights) {
|
}
|
||||||
| (Ok(distArray), Ok(w)) => Mixture(Belt.Array.zip(distArray, w)) -> runGenericOperation
|
| Some(EvDistribution(b)) => switch parseDistributionArray(args) {
|
||||||
| (Error(err), _) => GenDistError(ArgumentError(err))
|
| Ok(distributions) => mixtureWithDefaultWeights(distributions)
|
||||||
| (_, Error(err)) => GenDistError(ArgumentError(err))
|
| Error(err) => GenDistError(ArgumentError(err))
|
||||||
|
}
|
||||||
|
| _ => GenDistError(ArgumentError("Last argument of mx must be array or distribution"))
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
@ -178,8 +204,7 @@ let dispatchToGenericOutput = (call: ExpressionValue.functionCall): option<
|
||||||
Helpers.toDistFn(Truncate(None, Some(float)), dist)
|
Helpers.toDistFn(Truncate(None, Some(float)), dist)
|
||||||
| ("truncate", [EvDistribution(dist), EvNumber(float1), EvNumber(float2)]) =>
|
| ("truncate", [EvDistribution(dist), EvNumber(float1), EvNumber(float2)]) =>
|
||||||
Helpers.toDistFn(Truncate(Some(float1), Some(float2)), dist)
|
Helpers.toDistFn(Truncate(Some(float1), Some(float2)), dist)
|
||||||
| (("mx" | "mixture"), args) =>
|
| ("mx" | "mixture", args) => Helpers.mixture(args)->Some
|
||||||
Helpers.mixture(args) -> Some
|
|
||||||
| ("log", [EvDistribution(a)]) =>
|
| ("log", [EvDistribution(a)]) =>
|
||||||
Helpers.twoDiststoDistFn(Algebraic, "log", a, GenericDist.fromFloat(Math.e))->Some
|
Helpers.twoDiststoDistFn(Algebraic, "log", a, GenericDist.fromFloat(Math.e))->Some
|
||||||
| ("log10", [EvDistribution(a)]) =>
|
| ("log10", [EvDistribution(a)]) =>
|
||||||
|
@ -221,9 +246,8 @@ let genericOutputToReducerValue = (o: DistributionOperation.outputType): result<
|
||||||
| GenDistError(NotYetImplemented) => Error(RETodo("Function not yet implemented"))
|
| GenDistError(NotYetImplemented) => Error(RETodo("Function not yet implemented"))
|
||||||
| GenDistError(Unreachable) => Error(RETodo("Unreachable"))
|
| GenDistError(Unreachable) => Error(RETodo("Unreachable"))
|
||||||
| GenDistError(DistributionVerticalShiftIsInvalid) =>
|
| GenDistError(DistributionVerticalShiftIsInvalid) =>
|
||||||
Error(RETodo("Distribution Vertical Shift Is Invalid"))
|
Error(RETodo("Distribution Vertical Shift Is Invalid"))
|
||||||
| GenDistError(ArgumentError(err)) =>
|
| GenDistError(ArgumentError(err)) => Error(RETodo("Argument Error: " ++ err))
|
||||||
Error(RETodo("Argument Error: " ++ err))
|
|
||||||
| GenDistError(Other(s)) => Error(RETodo(s))
|
| GenDistError(Other(s)) => Error(RETodo(s))
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|
Loading…
Reference in New Issue
Block a user