API

Here are some examples of the machine learning API applied to regression models.

from __future__ import annotations

from gemseo import configure_logger
from gemseo import create_design_space
from gemseo import create_discipline
from gemseo import create_scenario
from gemseo.mlearning import create_regression_model
from gemseo.mlearning import get_regression_models
from gemseo.mlearning import get_regression_options

configure_logger()
<RootLogger root (INFO)>

Get available regression models

get_regression_models()
['GaussianProcessRegressor', 'LinearRegressor', 'MOERegressor', 'PCERegressor', 'PolynomialRegressor', 'RBFRegressor', 'RandomForestRegressor']

Get regression model options

get_regression_options("GaussianProcessRegressor")
+---------------------------+--------------------------------------------------------------------------------------------+---------------------------+
|            Name           |                                        Description                                         |            Type           |
+---------------------------+--------------------------------------------------------------------------------------------+---------------------------+
|           alpha           |                         The nugget effect to regularize the model.                         |           number          |
|           bounds          |   The lower and upper bounds of the parameter length scales when ``kernel`` is ``none``.   |            null           |
|                           | either a unique lower-upper pair common to all the inputs or lower-upper pairs for some of |                           |
|                           |  them. when ``bounds`` is ``none`` or when an input has no pair, the lower bound is 0.01   |                           |
|                           |                                and the upper bound is 100.                                 |                           |
|        input_names        |   The names of the input variables. if ``none``, consider all the input variables of the   |            null           |
|                           |                                     learning dataset.                                      |                           |
|           kernel          |        The kernel specifying the covariance model. if ``none``, use a matérn(2.5).         |            null           |
|    n_restarts_optimizer   |                          The number of restarts of the optimizer.                          |          integer          |
|         optimizer         |              The optimization algorithm to find the parameter length scales.               |           string          |
|        output_names       |  The names of the output variables. if ``none``, consider all the output variables of the  |            null           |
|                           |                                     learning dataset.                                      |                           |
|        random_state       |  The random state passed to the random number generator. use an integer for reproducible   |          integer          |
|                           |                                          results.                                          |                           |
|        transformer        |           The strategies to transform the variables. the values are instances of           |           object          |
|                           |  :class:`.transformer` while the keys are the names of either the variables or the groups  |                           |
|                           | of variables, e.g. ``"inputs"`` or ``"outputs"`` in the case of the regression algorithms. |                           |
|                           | if a group is specified, the :class:`.transformer` will be applied to all the variables of |                           |
|                           |             this group. if :attr:`.identity`, do not transform the variables.              |                           |
+---------------------------+--------------------------------------------------------------------------------------------+---------------------------+
    INFO - 00:11:45: +---------------------------+--------------------------------------------------------------------------------------------+---------------------------+
    INFO - 00:11:45: |            Name           |                                        Description                                         |            Type           |
    INFO - 00:11:45: +---------------------------+--------------------------------------------------------------------------------------------+---------------------------+
    INFO - 00:11:45: |           alpha           |                         The nugget effect to regularize the model.                         |           number          |
    INFO - 00:11:45: |           bounds          |   The lower and upper bounds of the parameter length scales when ``kernel`` is ``none``.   |            null           |
    INFO - 00:11:45: |                           | either a unique lower-upper pair common to all the inputs or lower-upper pairs for some of |                           |
    INFO - 00:11:45: |                           |  them. when ``bounds`` is ``none`` or when an input has no pair, the lower bound is 0.01   |                           |
    INFO - 00:11:45: |                           |                                and the upper bound is 100.                                 |                           |
    INFO - 00:11:45: |        input_names        |   The names of the input variables. if ``none``, consider all the input variables of the   |            null           |
    INFO - 00:11:45: |                           |                                     learning dataset.                                      |                           |
    INFO - 00:11:45: |           kernel          |        The kernel specifying the covariance model. if ``none``, use a matérn(2.5).         |            null           |
    INFO - 00:11:45: |    n_restarts_optimizer   |                          The number of restarts of the optimizer.                          |          integer          |
    INFO - 00:11:45: |         optimizer         |              The optimization algorithm to find the parameter length scales.               |           string          |
    INFO - 00:11:45: |        output_names       |  The names of the output variables. if ``none``, consider all the output variables of the  |            null           |
    INFO - 00:11:45: |                           |                                     learning dataset.                                      |                           |
    INFO - 00:11:45: |        random_state       |  The random state passed to the random number generator. use an integer for reproducible   |          integer          |
    INFO - 00:11:45: |                           |                                          results.                                          |                           |
    INFO - 00:11:45: |        transformer        |           The strategies to transform the variables. the values are instances of           |           object          |
    INFO - 00:11:45: |                           |  :class:`.transformer` while the keys are the names of either the variables or the groups  |                           |
    INFO - 00:11:45: |                           | of variables, e.g. ``"inputs"`` or ``"outputs"`` in the case of the regression algorithms. |                           |
    INFO - 00:11:45: |                           | if a group is specified, the :class:`.transformer` will be applied to all the variables of |                           |
    INFO - 00:11:45: |                           |             this group. if :attr:`.identity`, do not transform the variables.              |                           |
    INFO - 00:11:45: +---------------------------+--------------------------------------------------------------------------------------------+---------------------------+

{'$schema': 'http://json-schema.org/schema#', 'type': 'object', 'properties': {'transformer': {'description': 'The strategies to transform the variables. The values are instances of :class:`.Transformer` while the keys are the names of either the variables or the groups of variables, e.g. ``"inputs"`` or ``"outputs"`` in the case of the regression algorithms. If a group is specified, the :class:`.Transformer` will be applied to all the variables of this group. If :attr:`.IDENTITY`, do not transform the variables.', 'type': 'object'}, 'input_names': {'description': 'The names of the input variables. If ``None``, consider all the input variables of the learning dataset.', 'type': 'null'}, 'output_names': {'description': 'The names of the output variables. If ``None``, consider all the output variables of the learning dataset.', 'type': 'null'}, 'kernel': {'description': 'The kernel specifying the covariance model. If ``None``, use a Matérn(2.5).', 'type': 'null'}, 'bounds': {'description': 'The lower and upper bounds of the parameter length scales when ``kernel`` is ``None``. Either a unique lower-upper pair common to all the inputs or lower-upper pairs for some of them. When ``bounds`` is ``None`` or when an input has no pair, the lower bound is 0.01 and the upper bound is 100.', 'type': 'null'}, 'alpha': {'description': 'The nugget effect to regularize the model.', 'type': 'number'}, 'optimizer': {'description': 'The optimization algorithm to find the parameter length scales.', 'type': 'string'}, 'n_restarts_optimizer': {'description': 'The number of restarts of the optimizer.', 'type': 'integer'}, 'random_state': {'description': 'The random state passed to the random number generator. Use an integer for reproducible results.', 'type': 'integer'}}}

Create regression model

expressions = {"y_1": "1+2*x_1+3*x_2", "y_2": "-1-2*x_1-3*x_2"}
discipline = create_discipline(
    "AnalyticDiscipline", name="func", expressions=expressions
)

design_space = create_design_space()
design_space.add_variable("x_1", l_b=0.0, u_b=1.0)
design_space.add_variable("x_2", l_b=0.0, u_b=1.0)

scenario = create_scenario(
    [discipline], "DisciplinaryOpt", "y_1", design_space, scenario_type="DOE"
)
scenario.execute({"algo": "fullfact", "n_samples": 9})

dataset = scenario.to_dataset(opt_naming=False)
model = create_regression_model("LinearRegressor", data=dataset)
model.learn()
model
INFO - 00:11:45:
INFO - 00:11:45: *** Start DOEScenario execution ***
INFO - 00:11:45: DOEScenario
INFO - 00:11:45:    Disciplines: func
INFO - 00:11:45:    MDO formulation: DisciplinaryOpt
INFO - 00:11:45: Optimization problem:
INFO - 00:11:45:    minimize y_1(x_1, x_2)
INFO - 00:11:45:    with respect to x_1, x_2
INFO - 00:11:45:    over the design space:
INFO - 00:11:45:       +------+-------------+-------+-------------+-------+
INFO - 00:11:45:       | Name | Lower bound | Value | Upper bound | Type  |
INFO - 00:11:45:       +------+-------------+-------+-------------+-------+
INFO - 00:11:45:       | x_1  |      0      |  None |      1      | float |
INFO - 00:11:45:       | x_2  |      0      |  None |      1      | float |
INFO - 00:11:45:       +------+-------------+-------+-------------+-------+
INFO - 00:11:45: Solving optimization problem with algorithm fullfact:
INFO - 00:11:45:     11%|█         | 1/9 [00:00<00:00, 355.12 it/sec, obj=1]
INFO - 00:11:45:     22%|██▏       | 2/9 [00:00<00:00, 579.60 it/sec, obj=2]
INFO - 00:11:45:     33%|███▎      | 3/9 [00:00<00:00, 742.97 it/sec, obj=3]
INFO - 00:11:45:     44%|████▍     | 4/9 [00:00<00:00, 870.59 it/sec, obj=2.5]
INFO - 00:11:45:     56%|█████▌    | 5/9 [00:00<00:00, 962.92 it/sec, obj=3.5]
INFO - 00:11:45:     67%|██████▋   | 6/9 [00:00<00:00, 1043.27 it/sec, obj=4.5]
INFO - 00:11:45:     78%|███████▊  | 7/9 [00:00<00:00, 1110.70 it/sec, obj=4]
INFO - 00:11:45:     89%|████████▉ | 8/9 [00:00<00:00, 1167.60 it/sec, obj=5]
INFO - 00:11:45:    100%|██████████| 9/9 [00:00<00:00, 1216.01 it/sec, obj=6]
INFO - 00:11:45: Optimization result:
INFO - 00:11:45:    Optimizer info:
INFO - 00:11:45:       Status: None
INFO - 00:11:45:       Message: None
INFO - 00:11:45:       Number of calls to the objective function by the optimizer: 9
INFO - 00:11:45:    Solution:
INFO - 00:11:45:       Objective: 1.0
INFO - 00:11:45:       Design space:
INFO - 00:11:45:          +------+-------------+-------+-------------+-------+
INFO - 00:11:45:          | Name | Lower bound | Value | Upper bound | Type  |
INFO - 00:11:45:          +------+-------------+-------+-------------+-------+
INFO - 00:11:45:          | x_1  |      0      |   0   |      1      | float |
INFO - 00:11:45:          | x_2  |      0      |   0   |      1      | float |
INFO - 00:11:45:          +------+-------------+-------+-------------+-------+
INFO - 00:11:45: *** End DOEScenario execution (time: 0:00:00.018548) ***
LinearRegressor(fit_intercept=True, l2_penalty_ratio=1.0, penalty_level=0.0, random_state=0)
  • based on the scikit-learn library
  • built from 9 learning samples


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

Gallery generated by Sphinx-Gallery