Note
Go to the end to download the full example code
MDF-based MDO on the Sobieski SSBJ test case¶
from __future__ import annotations
from gemseo import configure_logger
from gemseo import create_discipline
from gemseo import create_scenario
from gemseo import generate_n2_plot
from gemseo.problems.sobieski.core.problem import SobieskiProblem
configure_logger()
<RootLogger root (INFO)>
Instantiate the disciplines¶
First, we instantiate the four disciplines of the use case:
SobieskiPropulsion
,
SobieskiAerodynamics
,
SobieskiMission
and SobieskiStructure
.
disciplines = create_discipline(
[
"SobieskiPropulsion",
"SobieskiAerodynamics",
"SobieskiMission",
"SobieskiStructure",
]
)
We can quickly access the most relevant information of any discipline (name, inputs,
and outputs) with Python’s print()
function. Moreover, we can get the default
input values of a discipline with the attribute MDODiscipline.default_inputs
for discipline in disciplines:
print(discipline)
print(f"Default inputs: {discipline.default_inputs}")
SobieskiPropulsion
Default inputs: {'x_3': array([0.5]), 'x_shared': array([5.0e-02, 4.5e+04, 1.6e+00, 5.5e+00, 5.5e+01, 1.0e+03]), 'y_23': array([12562.01206488]), 'c_3': array([4360.])}
SobieskiAerodynamics
Default inputs: {'x_2': array([1.]), 'x_shared': array([5.0e-02, 4.5e+04, 1.6e+00, 5.5e+00, 5.5e+01, 1.0e+03]), 'y_12': array([5.06069742e+04, 9.50000000e-01]), 'c_4': array([0.01375]), 'y_32': array([0.50279625])}
SobieskiMission
Default inputs: {'y_14': array([50606.9741711 , 7306.20262124]), 'x_shared': array([5.0e-02, 4.5e+04, 1.6e+00, 5.5e+00, 5.5e+01, 1.0e+03]), 'y_24': array([4.15006276]), 'y_34': array([1.10754577])}
SobieskiStructure
Default inputs: {'y_31': array([6354.32430691]), 'x_shared': array([5.0e-02, 4.5e+04, 1.6e+00, 5.5e+00, 5.5e+01, 1.0e+03]), 'c_0': array([2000.]), 'x_1': array([0.25, 1. ]), 'c_2': array([6.]), 'c_1': array([25000.]), 'y_21': array([50606.9741711])}
You may also be interested in plotting the couplings of your disciplines.
A quick way of getting this information is the high-level function
generate_n2_plot()
. A much more detailed explanation of coupling
visualization is available here.
generate_n2_plot(disciplines, save=False, show=True)
Build, execute and post-process the scenario¶
Then, we build the scenario which links the disciplines
with the formulation and the optimization algorithm. Here, we use the
MDF
formulation. We tell the scenario to minimize -y_4 instead of
minimizing y_4 (range), which is the default option.
Instantiate the scenario¶
During the instantiation of the scenario, we provide some options for the MDF formulations:
formulation_options = {
"tolerance": 1e-14,
"max_mda_iter": 50,
"warm_start": True,
"use_lu_fact": False,
"linear_solver_tolerance": 1e-14,
}
'warm_start
: warm starts MDA,'warm_start
: optimize the adjoints resolution by storing the Jacobian matrix LU factorization for the multiple RHS (objective + constraints). This saves CPU time if you can pay for the memory and have the full Jacobians available, not just matrix vector products.'linear_solver_tolerance'
: set the linear solver tolerance, idem we need full convergence
design_space = SobieskiProblem().design_space
print(design_space)
scenario = create_scenario(
disciplines,
"MDF",
objective_name="y_4",
design_space=design_space,
maximize_objective=True,
**formulation_options,
)
Design space:
+-------------+-------------+--------------------+-------------+-------+
| name | lower_bound | value | upper_bound | type |
+-------------+-------------+--------------------+-------------+-------+
| x_shared[0] | 0.01 | 0.05 | 0.09 | float |
| x_shared[1] | 30000 | 45000 | 60000 | float |
| x_shared[2] | 1.4 | 1.6 | 1.8 | float |
| x_shared[3] | 2.5 | 5.5 | 8.5 | float |
| x_shared[4] | 40 | 55 | 70 | float |
| x_shared[5] | 500 | 1000 | 1500 | float |
| x_1[0] | 0.1 | 0.25 | 0.4 | float |
| x_1[1] | 0.75 | 1 | 1.25 | float |
| x_2 | 0.75 | 1 | 1.25 | float |
| x_3 | 0.1 | 0.5 | 1 | float |
| y_14[0] | 24850 | 50606.9741711 | 77100 | float |
| y_14[1] | -7700 | 7306.20262124 | 45000 | float |
| y_32 | 0.235 | 0.5027962499999999 | 0.795 | float |
| y_31 | 2960 | 6354.32430691 | 10185 | float |
| y_24 | 0.44 | 4.15006276 | 11.13 | float |
| y_34 | 0.44 | 1.10754577 | 1.98 | float |
| y_23 | 3365 | 12194.2671934 | 26400 | float |
| y_21 | 24850 | 50606.9741711 | 77250 | float |
| y_12[0] | 24850 | 50606.9742 | 77250 | float |
| y_12[1] | 0.45 | 0.95 | 1.5 | float |
+-------------+-------------+--------------------+-------------+-------+
Set the design constraints¶
for c_name in ["g_1", "g_2", "g_3"]:
scenario.add_constraint(c_name, "ineq")
XDSMIZE the scenario¶
Generate the XDSM file on the fly:
log_workflow_status=True
will log the status of the workflow in the console,save_html
(defaultTrue
) will generate a self-contained HTML file, that can be automatically opened usingshow_html=True
.
scenario.xdsmize(save_html=False)
Define the algorithm inputs¶
We set the maximum number of iterations, the optimizer
and the optimizer options. Algorithm specific options are passed there.
Use the high-level function get_algorithm_options_schema()
for more information or read the documentation.
Here ftol_rel option is a stop criteria based on the relative difference in the objective between two iterates ineq_tolerance the tolerance determination of the optimum; this is specific to the GEMSEO wrapping and not in the solver.
algo_options = {
"ftol_rel": 1e-10,
"ineq_tolerance": 2e-3,
"normalize_design_space": True,
}
scn_inputs = {"max_iter": 15, "algo": "SLSQP", "algo_options": algo_options}
See also
We can also generate a backup file for the optimization,
as well as plots on the fly of the optimization history if option
generate_opt_plot
is True
.
This slows down a lot the process, here since SSBJ is very light
scenario.set_optimization_history_backup(file_path="mdf_backup.h5",
each_new_iter=True,
each_store=False, erase=True,
pre_load=False,
generate_opt_plot=True)
Execute the scenario¶
scenario.execute(scn_inputs)
INFO - 08:38:31:
INFO - 08:38:31: *** Start MDOScenario execution ***
INFO - 08:38:31: MDOScenario
INFO - 08:38:31: Disciplines: SobieskiAerodynamics SobieskiMission SobieskiPropulsion SobieskiStructure
INFO - 08:38:31: MDO formulation: MDF
INFO - 08:38:31: Optimization problem:
INFO - 08:38:31: minimize -y_4(x_shared, x_1, x_2, x_3)
INFO - 08:38:31: with respect to x_1, x_2, x_3, x_shared
INFO - 08:38:31: subject to constraints:
INFO - 08:38:31: g_1(x_shared, x_1, x_2, x_3) <= 0.0
INFO - 08:38:31: g_2(x_shared, x_1, x_2, x_3) <= 0.0
INFO - 08:38:31: g_3(x_shared, x_1, x_2, x_3) <= 0.0
INFO - 08:38:31: over the design space:
INFO - 08:38:31: +-------------+-------------+-------+-------------+-------+
INFO - 08:38:31: | name | lower_bound | value | upper_bound | type |
INFO - 08:38:31: +-------------+-------------+-------+-------------+-------+
INFO - 08:38:31: | x_shared[0] | 0.01 | 0.05 | 0.09 | float |
INFO - 08:38:31: | x_shared[1] | 30000 | 45000 | 60000 | float |
INFO - 08:38:31: | x_shared[2] | 1.4 | 1.6 | 1.8 | float |
INFO - 08:38:31: | x_shared[3] | 2.5 | 5.5 | 8.5 | float |
INFO - 08:38:31: | x_shared[4] | 40 | 55 | 70 | float |
INFO - 08:38:31: | x_shared[5] | 500 | 1000 | 1500 | float |
INFO - 08:38:31: | x_1[0] | 0.1 | 0.25 | 0.4 | float |
INFO - 08:38:31: | x_1[1] | 0.75 | 1 | 1.25 | float |
INFO - 08:38:31: | x_2 | 0.75 | 1 | 1.25 | float |
INFO - 08:38:31: | x_3 | 0.1 | 0.5 | 1 | float |
INFO - 08:38:31: +-------------+-------------+-------+-------------+-------+
INFO - 08:38:31: Solving optimization problem with algorithm SLSQP:
INFO - 08:38:31: ... 0%| | 0/15 [00:00<?, ?it]
INFO - 08:38:31: ... 7%|▋ | 1/15 [00:00<00:01, 9.88 it/sec, obj=-536]
INFO - 08:38:31: ... 13%|█▎ | 2/15 [00:00<00:01, 7.41 it/sec, obj=-2.12e+3]
INFO - 08:38:31: ... 20%|██ | 3/15 [00:00<00:01, 6.02 it/sec, obj=-3.72e+3]
INFO - 08:38:32: ... 27%|██▋ | 4/15 [00:00<00:01, 6.08 it/sec, obj=-3.97e+3]
INFO - 08:38:32: Optimization result:
INFO - 08:38:32: Optimizer info:
INFO - 08:38:32: Status: 8
INFO - 08:38:32: Message: Positive directional derivative for linesearch
INFO - 08:38:32: Number of calls to the objective function by the optimizer: 5
INFO - 08:38:32: Solution:
INFO - 08:38:32: The solution is feasible.
INFO - 08:38:32: Objective: -3716.555963095829
INFO - 08:38:32: Standardized constraints:
INFO - 08:38:32: g_1 = [-0.01608807 -0.03194613 -0.04316738 -0.05095364 -0.05658344 -0.1380806
INFO - 08:38:32: -0.1019194 ]
INFO - 08:38:32: g_2 = -0.0005956359157315294
INFO - 08:38:32: g_3 = [-0.67076432 -0.32923568 -0.10429595 -0.183255 ]
INFO - 08:38:32: Design space:
INFO - 08:38:32: +-------------+-------------+---------------------+-------------+-------+
INFO - 08:38:32: | name | lower_bound | value | upper_bound | type |
INFO - 08:38:32: +-------------+-------------+---------------------+-------------+-------+
INFO - 08:38:32: | x_shared[0] | 0.01 | 0.05985109102106711 | 0.09 | float |
INFO - 08:38:32: | x_shared[1] | 30000 | 59785.5639131558 | 60000 | float |
INFO - 08:38:32: | x_shared[2] | 1.4 | 1.4 | 1.8 | float |
INFO - 08:38:32: | x_shared[3] | 2.5 | 2.540129655171779 | 8.5 | float |
INFO - 08:38:32: | x_shared[4] | 40 | 69.80684214977607 | 70 | float |
INFO - 08:38:32: | x_shared[5] | 500 | 1493.746505655324 | 1500 | float |
INFO - 08:38:32: | x_1[0] | 0.1 | 0.4 | 0.4 | float |
INFO - 08:38:32: | x_1[1] | 0.75 | 0.7530970468499044 | 1.25 | float |
INFO - 08:38:32: | x_2 | 0.75 | 0.7530625218813826 | 1.25 | float |
INFO - 08:38:32: | x_3 | 0.1 | 0.1411034879427379 | 1 | float |
INFO - 08:38:32: +-------------+-------------+---------------------+-------------+-------+
INFO - 08:38:32: *** End MDOScenario execution (time: 0:00:00.775193) ***
{'max_iter': 15, 'algo_options': {'ftol_rel': 1e-10, 'ineq_tolerance': 0.002, 'normalize_design_space': True}, 'algo': 'SLSQP'}
Save the optimization history¶
We can save the whole optimization problem and its history for further post processing:
scenario.save_optimization_history("mdf_history.h5", file_format="hdf5")
INFO - 08:38:32: Export optimization problem to file: mdf_history.h5
We can also save only calls to functions and design variables history:
scenario.save_optimization_history("mdf_history.xml", file_format="ggobi")
INFO - 08:38:32: Export to ggobi for functions: ['-y_4', 'g_1', 'g_2', 'g_3']
INFO - 08:38:32: Export to ggobi file: mdf_history.xml
Print optimization metrics¶
scenario.print_execution_metrics()
INFO - 08:38:32: Scenario Execution Statistics
INFO - 08:38:32: Discipline: SobieskiPropulsion
INFO - 08:38:32: Executions number: 68
INFO - 08:38:32: Execution time: 0.018421720000333153 s
INFO - 08:38:32: Linearizations number: 4
INFO - 08:38:32: Discipline: SobieskiAerodynamics
INFO - 08:38:32: Executions number: 71
INFO - 08:38:32: Execution time: 0.025925376000486722 s
INFO - 08:38:32: Linearizations number: 4
INFO - 08:38:32: Discipline: SobieskiMission
INFO - 08:38:32: Executions number: 4
INFO - 08:38:32: Execution time: 0.0003651289998742868 s
INFO - 08:38:32: Linearizations number: 4
INFO - 08:38:32: Discipline: SobieskiStructure
INFO - 08:38:32: Executions number: 70
INFO - 08:38:32: Execution time: 0.09663684999668476 s
INFO - 08:38:32: Linearizations number: 4
INFO - 08:38:32: Total number of executions calls: 213
INFO - 08:38:32: Total number of linearizations: 16
Post-process the results¶
Plot the optimization history view¶
scenario.post_process("OptHistoryView", save=False, show=True)
/home/docs/checkouts/readthedocs.org/user_builds/gemseo/envs/5.0.1/lib/python3.9/site-packages/genson/schema/strategies/base.py:42: UserWarning: Schema incompatible. Keyword 'description' has conflicting values ('The width and height of the figure in inches, e.g. ``(w, h)``.\nIf ``None``, use the :attr:`.OptPostProcessor.DEFAULT_FIG_SIZE`\nof the post-processor.' vs. 'The width and height of the figure in inches, e.g. `(w, h)`.\nIf ``None``, use the :attr:`.OptPostProcessor.DEFAULT_FIG_SIZE`\nof the post-processor.'). Using 'The width and height of the figure in inches, e.g. ``(w, h)``.\nIf ``None``, use the :attr:`.OptPostProcessor.DEFAULT_FIG_SIZE`\nof the post-processor.'
warn(('Schema incompatible. Keyword {0!r} has conflicting '
<gemseo.post.opt_history_view.OptHistoryView object at 0x7f8736587160>
Plot the basic history view¶
scenario.post_process(
"BasicHistory", variable_names=["x_shared"], save=False, show=True
)
<gemseo.post.basic_history.BasicHistory object at 0x7f8733fc3610>
Plot the constraints and objective history¶
scenario.post_process("ObjConstrHist", save=False, show=True)
/home/docs/checkouts/readthedocs.org/user_builds/gemseo/envs/5.0.1/lib/python3.9/site-packages/genson/schema/strategies/base.py:42: UserWarning: Schema incompatible. Keyword 'description' has conflicting values ('The width and height of the figure in inches, e.g. ``(w, h)``.\nIf ``None``, use the :attr:`.OptPostProcessor.DEFAULT_FIG_SIZE`\nof the post-processor.' vs. 'The width and height of the figure in inches, e.g. `(w, h)`.\nIf ``None``, use the :attr:`.OptPostProcessor.DEFAULT_FIG_SIZE`\nof the post-processor.'). Using 'The width and height of the figure in inches, e.g. ``(w, h)``.\nIf ``None``, use the :attr:`.OptPostProcessor.DEFAULT_FIG_SIZE`\nof the post-processor.'
warn(('Schema incompatible. Keyword {0!r} has conflicting '
<gemseo.post.obj_constr_hist.ObjConstrHist object at 0x7f8733c29b50>
Plot the constraints history¶
scenario.post_process(
"ConstraintsHistory",
constraint_names=["g_1", "g_2", "g_3"],
save=False,
show=True,
)
<gemseo.post.constraints_history.ConstraintsHistory object at 0x7f8733a1f2b0>
Plot the constraints history using a radar chart¶
scenario.post_process(
"RadarChart",
constraint_names=["g_1", "g_2", "g_3"],
save=False,
show=True,
)
<gemseo.post.radar_chart.RadarChart object at 0x7f8733bb0af0>
Plot the quadratic approximation of the objective¶
scenario.post_process("QuadApprox", function="-y_4", save=False, show=True)
/home/docs/checkouts/readthedocs.org/user_builds/gemseo/envs/5.0.1/lib/python3.9/site-packages/genson/schema/strategies/base.py:42: UserWarning: Schema incompatible. Keyword 'description' has conflicting values ('The width and height of the figure in inches, e.g. ``(w, h)``.\nIf ``None``, use the :attr:`.OptPostProcessor.DEFAULT_FIG_SIZE`\nof the post-processor.' vs. 'The width and height of the figure in inches, e.g. `(w, h)`.\nIf ``None``, use the :attr:`.OptPostProcessor.DEFAULT_FIG_SIZE`\nof the post-processor.'). Using 'The width and height of the figure in inches, e.g. ``(w, h)``.\nIf ``None``, use the :attr:`.OptPostProcessor.DEFAULT_FIG_SIZE`\nof the post-processor.'
warn(('Schema incompatible. Keyword {0!r} has conflicting '
<gemseo.post.quad_approx.QuadApprox object at 0x7f87334edca0>
Plot the functions using a SOM¶
scenario.post_process("SOM", save=False, show=True)
INFO - 08:38:36: Building Self Organizing Map from optimization history:
INFO - 08:38:36: Number of neurons in x direction = 4
INFO - 08:38:36: Number of neurons in y direction = 4
<gemseo.post.som.SOM object at 0x7f8733437370>
Plot the scatter matrix of variables of interest¶
scenario.post_process(
"ScatterPlotMatrix",
variable_names=["-y_4", "g_1"],
save=False,
show=True,
fig_size=(14, 14),
)
/home/docs/checkouts/readthedocs.org/user_builds/gemseo/envs/5.0.1/lib/python3.9/site-packages/genson/schema/strategies/base.py:42: UserWarning: Schema incompatible. Keyword 'description' has conflicting values ('The width and height of the figure in inches, e.g. ``(w, h)``.\nIf ``None``, use the :attr:`.OptPostProcessor.DEFAULT_FIG_SIZE`\nof the post-processor.' vs. 'The width and height of the figure in inches, e.g. `(w, h)`.\nIf ``None``, use the :attr:`.OptPostProcessor.DEFAULT_FIG_SIZE`\nof the post-processor.'). Using 'The width and height of the figure in inches, e.g. ``(w, h)``.\nIf ``None``, use the :attr:`.OptPostProcessor.DEFAULT_FIG_SIZE`\nof the post-processor.'
warn(('Schema incompatible. Keyword {0!r} has conflicting '
<gemseo.post.scatter_mat.ScatterPlotMatrix object at 0x7f873286afd0>
Plot the variables using the parallel coordinates¶
scenario.post_process("ParallelCoordinates", save=False, show=True)
/home/docs/checkouts/readthedocs.org/user_builds/gemseo/envs/5.0.1/lib/python3.9/site-packages/genson/schema/strategies/base.py:42: UserWarning: Schema incompatible. Keyword 'description' has conflicting values ('The width and height of the figure in inches, e.g. ``(w, h)``.\nIf ``None``, use the :attr:`.OptPostProcessor.DEFAULT_FIG_SIZE`\nof the post-processor.' vs. 'The width and height of the figure in inches, e.g. `(w, h)`.\nIf ``None``, use the :attr:`.OptPostProcessor.DEFAULT_FIG_SIZE`\nof the post-processor.'). Using 'The width and height of the figure in inches, e.g. ``(w, h)``.\nIf ``None``, use the :attr:`.OptPostProcessor.DEFAULT_FIG_SIZE`\nof the post-processor.'
warn(('Schema incompatible. Keyword {0!r} has conflicting '
<gemseo.post.para_coord.ParallelCoordinates object at 0x7f8733ac3040>
Plot the robustness of the solution¶
scenario.post_process("Robustness", save=True, show=True)
<gemseo.post.robustness.Robustness object at 0x7f87328d7280>
Plot the influence of the design variables¶
scenario.post_process("VariableInfluence", fig_size=(14, 14), save=False, show=True)
/home/docs/checkouts/readthedocs.org/user_builds/gemseo/envs/5.0.1/lib/python3.9/site-packages/genson/schema/strategies/base.py:42: UserWarning: Schema incompatible. Keyword 'description' has conflicting values ('The width and height of the figure in inches, e.g. ``(w, h)``.\nIf ``None``, use the :attr:`.OptPostProcessor.DEFAULT_FIG_SIZE`\nof the post-processor.' vs. 'The width and height of the figure in inches, e.g. `(w, h)`.\nIf ``None``, use the :attr:`.OptPostProcessor.DEFAULT_FIG_SIZE`\nof the post-processor.'). Using 'The width and height of the figure in inches, e.g. ``(w, h)``.\nIf ``None``, use the :attr:`.OptPostProcessor.DEFAULT_FIG_SIZE`\nof the post-processor.'
warn(('Schema incompatible. Keyword {0!r} has conflicting '
INFO - 08:38:42: Output name; most influential variables to explain 0.99% of the output variation
INFO - 08:38:42: -y_4; x_1[1], x_2, x_3, x_shared[0], x_shared[1], x_shared[2], x_shared[3], x_shared[4], x_shared[5]
/home/docs/checkouts/readthedocs.org/user_builds/gemseo/envs/5.0.1/lib/python3.9/site-packages/gemseo/post/variable_influence.py:238: UserWarning: FixedFormatter should only be used together with FixedLocator
axe.set_xticklabels(x_labels, fontsize=font_size, rotation=rotation)
INFO - 08:38:42: g_1[0]; x_1[0], x_1[1], x_shared[0], x_shared[3], x_shared[5]
INFO - 08:38:42: g_1[1]; x_1[0], x_1[1], x_shared[0], x_shared[3], x_shared[5]
INFO - 08:38:42: g_1[2]; x_1[0], x_1[1], x_shared[0], x_shared[3], x_shared[5]
INFO - 08:38:42: g_1[3]; x_1[0], x_1[1], x_shared[0], x_shared[3], x_shared[5]
INFO - 08:38:42: g_1[4]; x_1[0], x_1[1], x_shared[0], x_shared[3], x_shared[5]
INFO - 08:38:42: g_1[5]; x_1[0], x_1[1], x_shared[3], x_shared[5]
INFO - 08:38:42: g_1[6]; x_1[0], x_1[1], x_shared[3], x_shared[5]
INFO - 08:38:42: g_2; x_shared[0]
INFO - 08:38:42: g_3[0]; x_2, x_3, x_shared[0], x_shared[1], x_shared[2], x_shared[4], x_shared[5]
INFO - 08:38:42: g_3[1]; x_2, x_3, x_shared[0], x_shared[1], x_shared[2], x_shared[4], x_shared[5]
INFO - 08:38:42: g_3[2]; x_3, x_shared[1], x_shared[2]
INFO - 08:38:42: g_3[3]; x_3, x_shared[1], x_shared[2]
<gemseo.post.variable_influence.VariableInfluence object at 0x7f87334cf3a0>
Total running time of the script: ( 0 minutes 12.245 seconds)