1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103
|
#! /usr/bin/env python
import openturns as ot
from openturns.usecases import flood_model
ot.TESTPREAMBLE()
ot.Log.Show(ot.Log.NONE)
sampleSize = 500
totalDegree = 7
#
print("+ Compute flood model using least squares")
fm = flood_model.FloodModel()
inputDescription = fm.model.getInputDescription()
marginals = [fm.distribution.getMarginal(i) for i in range(fm.dim)]
basis = ot.OrthogonalProductPolynomialFactory(marginals)
inputSample = fm.distribution.getSample(sampleSize)
outputSample = fm.model(inputSample)
selectionAlgorithm = ot.LeastSquaresMetaModelSelectionFactory()
leastSquaresStrategy = ot.LeastSquaresStrategy(selectionAlgorithm)
print("leastSquaresStrategy")
print(leastSquaresStrategy)
print("leastSquaresStrategy (repr)")
print(leastSquaresStrategy.__repr__())
print("leastSquaresStrategy (html)")
print(leastSquaresStrategy._repr_html_())
assert leastSquaresStrategy.isLeastSquares()
assert leastSquaresStrategy.involvesModelSelection()
assert selectionAlgorithm.involvesModelSelection()
enumerateFunction = basis.getEnumerateFunction()
basisSize = enumerateFunction.getBasisSizeFromTotalDegree(totalDegree)
adaptiveStrategy = ot.FixedStrategy(basis, basisSize)
chaosAlgo = ot.FunctionalChaosAlgorithm(
inputSample, outputSample, fm.distribution, adaptiveStrategy, leastSquaresStrategy
)
projectionStrategy = chaosAlgo.getProjectionStrategy()
print("projectionStrategy")
print(projectionStrategy)
print("projectionStrategy (repr)")
print(projectionStrategy.__repr__())
print("projectionStrategy (html)")
print(projectionStrategy._repr_html_())
assert not adaptiveStrategy.involvesModelSelection()
#
print("+ Compute flood model by integration")
integrationStrategy = ot.IntegrationStrategy()
print("integrationStrategy")
print(integrationStrategy)
print("integrationStrategy (repr)")
print(integrationStrategy.__repr__())
print("integrationStrategy (html)")
print(integrationStrategy._repr_html_())
chaosAlgo = ot.FunctionalChaosAlgorithm(
inputSample, outputSample, fm.distribution, adaptiveStrategy, integrationStrategy
)
projectionStrategy = chaosAlgo.getProjectionStrategy()
print("projectionStrategy")
print(projectionStrategy)
print("projectionStrategy (repr)")
print(projectionStrategy.__repr__())
print("projectionStrategy (html)")
print(projectionStrategy._repr_html_())
assert not integrationStrategy.isLeastSquares()
#
print("+ Compute flood model with large output dimension")
outputDimension = 20
fm = flood_model.FloodModel()
model = ot.AggregatedFunction([fm.model] * outputDimension)
sampleSize = 500
totalDegree = 7
fm = flood_model.FloodModel()
inputDescription = fm.model.getInputDescription()
marginals = [fm.distribution.getMarginal(i) for i in range(fm.dim)]
basis = ot.OrthogonalProductPolynomialFactory(marginals)
inputSample = fm.distribution.getSample(sampleSize)
outputSample = model(inputSample)
selectionAlgorithm = ot.LeastSquaresMetaModelSelectionFactory()
leastSquaresStrategy = ot.LeastSquaresStrategy(selectionAlgorithm)
print("leastSquaresStrategy")
print(leastSquaresStrategy)
print("leastSquaresStrategy (repr)")
print(leastSquaresStrategy.__repr__())
print(leastSquaresStrategy._repr_html_())
enumerateFunction = basis.getEnumerateFunction()
basisSize = enumerateFunction.getBasisSizeFromTotalDegree(totalDegree)
adaptiveStrategy = ot.FixedStrategy(basis, basisSize)
chaosAlgo = ot.FunctionalChaosAlgorithm(
inputSample, outputSample, fm.distribution, adaptiveStrategy, leastSquaresStrategy
)
projectionStrategy = chaosAlgo.getProjectionStrategy()
print("projectionStrategy")
print(projectionStrategy)
print("projectionStrategy (repr)")
print(projectionStrategy.__repr__())
print("projectionStrategy (html)")
print(projectionStrategy._repr_html_())
|