IDF-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]), 'c_3': array([4360.]), '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])}
SobieskiAerodynamics
Default inputs: {'y_12': array([5.06069742e+04, 9.50000000e-01]), 'y_32': array([0.50279625]), 'x_2': array([1.]), 'c_4': array([0.01375]), 'x_shared': array([5.0e-02, 4.5e+04, 1.6e+00, 5.5e+00, 5.5e+01, 1.0e+03])}
SobieskiMission
Default inputs: {'y_24': array([4.15006276]), 'y_34': array([1.10754577]), 'x_shared': array([5.0e-02, 4.5e+04, 1.6e+00, 5.5e+00, 5.5e+01, 1.0e+03]), 'y_14': array([50606.9741711 ,  7306.20262124])}
SobieskiStructure
Default inputs: {'y_31': array([6354.32430691]), 'c_2': array([6.]), 'x_1': array([0.25, 1.  ]), 'y_21': array([50606.9741711]), 'c_1': array([25000.]), 'c_0': array([2000.]), 'x_shared': array([5.0e-02, 4.5e+04, 1.6e+00, 5.5e+00, 5.5e+01, 1.0e+03])}

You may also be interested in plotting the couplings of your disciplines. A quick way of getting this information is the API function generate_n2_plot(). A much more detailed explanation of coupling visualization is available here.

generate_n2_plot(disciplines, save=False, show=True)
plot sobieski idf example

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 IDF formulation. We tell the scenario to minimize -y_4 instead of minimizing y_4 (range), which is the default option.

Instantiate the scenario

design_space = SobieskiProblem().design_space
design_space
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


scenario = create_scenario(
    disciplines,
    "IDF",
    objective_name="y_4",
    design_space=design_space,
    maximize_objective=True,
)

Set the design constraints

for c_name in ["g_1", "g_2", "g_3"]:
    scenario.add_constraint(c_name, "ineq")

Visualize the XDSM

Generate the XDSM file on the fly:

  • log_workflow_status=True will log the status of the workflow in the console,

  • save_html (default True) will generate a self-contained HTML file, that can be automatically opened using show_html=True.

scenario.xdsmize(save_html=False)


Define the algorithm inputs

We set the maximum number of iterations, the optimizer and the optimizer options

algo_options = {
    "ftol_rel": 1e-10,
    "ineq_tolerance": 1e-3,
    "eq_tolerance": 1e-3,
    "normalize_design_space": True,
}
scn_inputs = {"max_iter": 20, "algo": "SLSQP", "algo_options": algo_options}

Execute the scenario

scenario.execute(scn_inputs)
    INFO - 08:49:23:
    INFO - 08:49:23: *** Start MDOScenario execution ***
    INFO - 08:49:23: MDOScenario
    INFO - 08:49:23:    Disciplines: SobieskiAerodynamics SobieskiMission SobieskiPropulsion SobieskiStructure
    INFO - 08:49:23:    MDO formulation: IDF
    INFO - 08:49:23: Optimization problem:
    INFO - 08:49:23:    minimize -y_4(x_shared, y_14, y_24, y_34)
    INFO - 08:49:23:    with respect to x_1, x_2, x_3, x_shared, y_12, y_14, y_21, y_23, y_24, y_31, y_32, y_34
    INFO - 08:49:23:    subject to constraints:
    INFO - 08:49:23:       g_1(x_shared, x_1, y_31, y_21) <= 0.0
    INFO - 08:49:23:       g_2(x_shared, x_2, y_32, y_12) <= 0.0
    INFO - 08:49:23:       g_3(x_shared, x_3, y_23) <= 0.0
    INFO - 08:49:23:       y_31_y_32_y_34(x_shared, x_3, y_23): y_31(x_shared, x_3, y_23) - y_31 == 0.0
    INFO - 08:49:23:                                            y_32(x_shared, x_3, y_23) - y_32 == 0.0
    INFO - 08:49:23:                                            y_34(x_shared, x_3, y_23) - y_34 == 0.0
    INFO - 08:49:23:       y_21_y_23_y_24(x_shared, x_2, y_32, y_12): y_21(x_shared, x_2, y_32, y_12) - y_21 == 0.0
    INFO - 08:49:23:                                                  y_23(x_shared, x_2, y_32, y_12) - y_23 == 0.0
    INFO - 08:49:23:                                                  y_24(x_shared, x_2, y_32, y_12) - y_24 == 0.0
    INFO - 08:49:23:       y_12_y_14(x_shared, x_1, y_31, y_21): y_12(x_shared, x_1, y_31, y_21) - y_12 == 0.0
    INFO - 08:49:23:                                             y_14(x_shared, x_1, y_31, y_21) - y_14 == 0.0
    INFO - 08:49:23:    over the design space:
    INFO - 08:49:23:    +-------------+-------------+--------------------+-------------+-------+
    INFO - 08:49:23:    | name        | lower_bound |       value        | upper_bound | type  |
    INFO - 08:49:23:    +-------------+-------------+--------------------+-------------+-------+
    INFO - 08:49:23:    | x_shared[0] |     0.01    |        0.05        |     0.09    | float |
    INFO - 08:49:23:    | x_shared[1] |    30000    |       45000        |    60000    | float |
    INFO - 08:49:23:    | x_shared[2] |     1.4     |        1.6         |     1.8     | float |
    INFO - 08:49:23:    | x_shared[3] |     2.5     |        5.5         |     8.5     | float |
    INFO - 08:49:23:    | x_shared[4] |      40     |         55         |      70     | float |
    INFO - 08:49:23:    | x_shared[5] |     500     |        1000        |     1500    | float |
    INFO - 08:49:23:    | x_1[0]      |     0.1     |        0.25        |     0.4     | float |
    INFO - 08:49:23:    | x_1[1]      |     0.75    |         1          |     1.25    | float |
    INFO - 08:49:23:    | x_2         |     0.75    |         1          |     1.25    | float |
    INFO - 08:49:23:    | x_3         |     0.1     |        0.5         |      1      | float |
    INFO - 08:49:23:    | y_14[0]     |    24850    |   50606.9741711    |    77100    | float |
    INFO - 08:49:23:    | y_14[1]     |    -7700    |   7306.20262124    |    45000    | float |
    INFO - 08:49:23:    | y_32        |    0.235    | 0.5027962499999999 |    0.795    | float |
    INFO - 08:49:23:    | y_31        |     2960    |   6354.32430691    |    10185    | float |
    INFO - 08:49:23:    | y_24        |     0.44    |     4.15006276     |    11.13    | float |
    INFO - 08:49:23:    | y_34        |     0.44    |     1.10754577     |     1.98    | float |
    INFO - 08:49:23:    | y_23        |     3365    |   12194.2671934    |    26400    | float |
    INFO - 08:49:23:    | y_21        |    24850    |   50606.9741711    |    77250    | float |
    INFO - 08:49:23:    | y_12[0]     |    24850    |     50606.9742     |    77250    | float |
    INFO - 08:49:23:    | y_12[1]     |     0.45    |        0.95        |     1.5     | float |
    INFO - 08:49:23:    +-------------+-------------+--------------------+-------------+-------+
    INFO - 08:49:23: Solving optimization problem with algorithm SLSQP:
    INFO - 08:49:23: ...   0%|          | 0/20 [00:00<?, ?it]
    INFO - 08:49:23: ...   5%|▌         | 1/20 [00:00<00:00, 253.39 it/sec, obj=-536]
    INFO - 08:49:23: ...  10%|█         | 2/20 [00:00<00:00, 66.71 it/sec, obj=-1.49e+3]
    INFO - 08:49:23: ...  15%|█▌        | 3/20 [00:00<00:00, 74.61 it/sec, obj=-3.83e+3]
    INFO - 08:49:23: ...  20%|██        | 4/20 [00:00<00:00, 79.13 it/sec, obj=-3.87e+3]
    INFO - 08:49:23: ...  25%|██▌       | 5/20 [00:00<00:00, 82.17 it/sec, obj=-3.96e+3]
    INFO - 08:49:23: ...  30%|███       | 6/20 [00:00<00:00, 84.29 it/sec, obj=-3.96e+3]
    INFO - 08:49:23: ...  35%|███▌      | 7/20 [00:00<00:00, 85.51 it/sec, obj=-3.96e+3]
    INFO - 08:49:23: ...  40%|████      | 8/20 [00:00<00:00, 92.09 it/sec, obj=-3.96e+3]
    INFO - 08:49:23: ...  45%|████▌     | 9/20 [00:00<00:00, 98.04 it/sec, obj=-3.96e+3]
    INFO - 08:49:23: ...  50%|█████     | 10/20 [00:00<00:00, 103.39 it/sec, obj=-3.96e+3]
    INFO - 08:49:23: ...  55%|█████▌    | 11/20 [00:00<00:00, 108.21 it/sec, obj=-3.96e+3]
    INFO - 08:49:23: ...  60%|██████    | 12/20 [00:00<00:00, 112.57 it/sec, obj=-3.96e+3]
    INFO - 08:49:23: ...  65%|██████▌   | 13/20 [00:00<00:00, 116.47 it/sec, obj=-3.96e+3]
    INFO - 08:49:23: ...  70%|███████   | 14/20 [00:00<00:00, 120.09 it/sec, obj=-3.96e+3]
    INFO - 08:49:23: ...  75%|███████▌  | 15/20 [00:00<00:00, 123.42 it/sec, obj=-3.96e+3]
    INFO - 08:49:23: ...  80%|████████  | 16/20 [00:00<00:00, 126.53 it/sec, obj=-3.96e+3]
    INFO - 08:49:23: ...  85%|████████▌ | 17/20 [00:00<00:00, 129.41 it/sec, obj=-3.96e+3]
    INFO - 08:49:23: ...  90%|█████████ | 18/20 [00:00<00:00, 126.79 it/sec, obj=-3.97e+3]
    INFO - 08:49:23: ...  95%|█████████▌| 19/20 [00:00<00:00, 129.31 it/sec, obj=-3.97e+3]
    INFO - 08:49:23: ... 100%|██████████| 20/20 [00:00<00:00, 131.60 it/sec, obj=-3.96e+3]
    INFO - 08:49:23: Optimization result:
    INFO - 08:49:23:    Optimizer info:
    INFO - 08:49:23:       Status: None
    INFO - 08:49:23:       Message: Maximum number of iterations reached. GEMSEO Stopped the driver
    INFO - 08:49:23:       Number of calls to the objective function by the optimizer: 22
    INFO - 08:49:23:    Solution:
    INFO - 08:49:23:       The solution is feasible.
    INFO - 08:49:23:       Objective: -3966.790848506069
    INFO - 08:49:23:       Standardized constraints:
    INFO - 08:49:23:          g_1 = [-0.01812782 -0.03339445 -0.0442868  -0.0518651  -0.05735184 -0.13720865
    INFO - 08:49:23:  -0.10279135]
    INFO - 08:49:23:          g_2 = 2.535682693616259e-05
    INFO - 08:49:23:          g_3 = [-7.67489138e-01 -2.32510862e-01  9.20908972e-05 -1.83255000e-01]
    INFO - 08:49:23:          y_12_y_14 = [ 3.34629122e-05 -1.85828412e-05  3.35589780e-05 -5.48378413e-05]
    INFO - 08:49:23:          y_21_y_23_y_24 = [ 0.00000000e+00  9.05477756e-05 -2.83060860e-04]
    INFO - 08:49:23:          y_31_y_32_y_34 = [6.90641576e-08 6.03563891e-08 1.07078689e-04]
    INFO - 08:49:23:       Design space:
    INFO - 08:49:23:       +-------------+-------------+---------------------+-------------+-------+
    INFO - 08:49:23:       | name        | lower_bound |        value        | upper_bound | type  |
    INFO - 08:49:23:       +-------------+-------------+---------------------+-------------+-------+
    INFO - 08:49:23:       | x_shared[0] |     0.01    | 0.06000633920673407 |     0.09    | float |
    INFO - 08:49:23:       | x_shared[1] |    30000    |        60000        |    60000    | float |
    INFO - 08:49:23:       | x_shared[2] |     1.4     |         1.4         |     1.8     | float |
    INFO - 08:49:23:       | x_shared[3] |     2.5     |         2.5         |     8.5     | float |
    INFO - 08:49:23:       | x_shared[4] |      40     |          70         |      70     | float |
    INFO - 08:49:23:       | x_shared[5] |     500     |         1500        |     1500    | float |
    INFO - 08:49:23:       | x_1[0]      |     0.1     |         0.4         |     0.4     | float |
    INFO - 08:49:23:       | x_1[1]      |     0.75    |         0.75        |     1.25    | float |
    INFO - 08:49:23:       | x_2         |     0.75    |         0.75        |     1.25    | float |
    INFO - 08:49:23:       | x_3         |     0.1     |  0.156259134365097  |      1      | float |
    INFO - 08:49:23:       | y_14[0]     |    24850    |  44746.56244377587  |    77100    | float |
    INFO - 08:49:23:       | y_14[1]     |    -7700    |  19355.27544458607  |    45000    | float |
    INFO - 08:49:23:       | y_32        |    0.235    |  0.7325108278699097 |    0.795    | float |
    INFO - 08:49:23:       | y_31        |     2960    |  9433.274956925387  |    10185    | float |
    INFO - 08:49:23:       | y_24        |     0.44    |  8.059470045805423  |    11.13    | float |
    INFO - 08:49:23:       | y_34        |     0.44    |  0.9237614560895575 |     1.98    | float |
    INFO - 08:49:23:       | y_23        |     3365    |  5552.047264915329  |    26400    | float |
    INFO - 08:49:23:       | y_21        |    24850    |  44746.56244377587  |    77250    | float |
    INFO - 08:49:23:       | y_12[0]     |    24850    |  44746.56244377587  |    77250    | float |
    INFO - 08:49:23:       | y_12[1]     |     0.45    |  0.9028108618065331 |     1.5     | float |
    INFO - 08:49:23:       +-------------+-------------+---------------------+-------------+-------+
    INFO - 08:49:23: *** End MDOScenario execution (time: 0:00:00.181274) ***

{'max_iter': 20, 'algo_options': {'ftol_rel': 1e-10, 'ineq_tolerance': 0.001, 'eq_tolerance': 0.001, '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("idf_history.h5", file_format="hdf5")
INFO - 08:49:23: Export optimization problem to file: idf_history.h5

We can also save only calls to functions and design variables history:

scenario.save_optimization_history("idf_history.xml", file_format="ggobi")
INFO - 08:49:23: Export to ggobi for functions: ['-y_4', 'g_1', 'g_2', 'g_3', 'y_12_y_14', 'y_21_y_23_y_24', 'y_31_y_32_y_34']
INFO - 08:49:23: Export to ggobi file: idf_history.xml

Plot the optimization history view

scenario.post_process("OptHistoryView", save=True, show=True)
<gemseo.post.opt_history_view.OptHistoryView object at 0x7f5db991d670>

Plot the quadratic approximation of the objective

scenario.post_process("QuadApprox", function="-y_4", save=False, show=True)
  • Hessian matrix SR1 approximation of -y_4
  • plot sobieski idf example
<gemseo.post.quad_approx.QuadApprox object at 0x7f5db88a0910>

Total running time of the script: ( 0 minutes 4.370 seconds)

Gallery generated by Sphinx-Gallery