Use a design of experiments from a file

from __future__ import annotations

from pathlib import Path

from gemseo import create_design_space
from gemseo import create_discipline
from gemseo import create_scenario

Let us consider a discipline implementing the function \(y=a*b\)

discipline = create_discipline("AnalyticDiscipline", expressions={"y": "a*b"})

where \(a,b\in\{1,2,\ldots,10\}\):

design_space = create_design_space()
design_space.add_variable("a", 1, design_space.DesignVariableType.INTEGER, 1, 10)
design_space.add_variable("b", 1, design_space.DesignVariableType.INTEGER, 1, 10)

We want to evaluate this discipline over this design space by using the input samples defined in the file "doe.txt":

print(Path("doe.txt").read_text())

1,2
3,4
5,6

In this file, rows are points and columns are variables whose order must be consistent with that of the design space. In this example, we can see that the first input value is defined by \(a=1\) and \(b=2\).

For that, we can create a scenario and execute it with a CustomDOE, with the option "doe_file". We could also change the delimiter (default: ',') or skip the first rows in the file.

scenario = create_scenario(
    [discipline],
    "y",
    design_space,
    scenario_type="DOE",
    formulation_name="DisciplinaryOpt",
)
scenario.execute(algo_name="CustomDOE", doe_file="doe.txt")

Note that both the formulation settings passed to create_scenario() and the algorithm settings passed to execute() can be provided via a Pydantic model. For more information, see Formulation Settings and Algorithm Settings.

We can display the content of the database as a Dataset and check the values of the output, which should be the product of \(a\) and \(b\).

opt_problem = scenario.formulation.optimization_problem
dataset = opt_problem.to_dataset(name="custom_sampling", opt_naming=False)
dataset

GROUP	inputs		outputs
VARIABLE	a	b	y
COMPONENT	0	0	0
0	1	2	2.0
1	3	4	12.0
2	5	6	30.0