Note
Go to the end to download the full example code
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 - 09:01:37: +---------------------------+--------------------------------------------------------------------------------------------+---------------------------+
INFO - 09:01:37: | Name | Description | Type |
INFO - 09:01:37: +---------------------------+--------------------------------------------------------------------------------------------+---------------------------+
INFO - 09:01:37: | alpha | The nugget effect to regularize the model. | number |
INFO - 09:01:37: | bounds | The lower and upper bounds of the parameter length scales when ``kernel`` is ``none``. | null |
INFO - 09:01:37: | | either a unique lower-upper pair common to all the inputs or lower-upper pairs for some of | |
INFO - 09:01:37: | | them. when ``bounds`` is ``none`` or when an input has no pair, the lower bound is 0.01 | |
INFO - 09:01:37: | | and the upper bound is 100. | |
INFO - 09:01:37: | input_names | The names of the input variables. if ``none``, consider all the input variables of the | null |
INFO - 09:01:37: | | learning dataset. | |
INFO - 09:01:37: | kernel | The kernel specifying the covariance model. if ``none``, use a matérn(2.5). | null |
INFO - 09:01:37: | n_restarts_optimizer | The number of restarts of the optimizer. | integer |
INFO - 09:01:37: | optimizer | The optimization algorithm to find the parameter length scales. | string |
INFO - 09:01:37: | output_names | The names of the output variables. if ``none``, consider all the output variables of the | null |
INFO - 09:01:37: | | learning dataset. | |
INFO - 09:01:37: | random_state | The random state passed to the random number generator. use an integer for reproducible | integer |
INFO - 09:01:37: | | results. | |
INFO - 09:01:37: | transformer | The strategies to transform the variables. the values are instances of | object |
INFO - 09:01:37: | | :class:`.transformer` while the keys are the names of either the variables or the groups | |
INFO - 09:01:37: | | of variables, e.g. ``"inputs"`` or ``"outputs"`` in the case of the regression algorithms. | |
INFO - 09:01:37: | | if a group is specified, the :class:`.transformer` will be applied to all the variables of | |
INFO - 09:01:37: | | this group. if :attr:`.identity`, do not transform the variables. | |
INFO - 09:01:37: +---------------------------+--------------------------------------------------------------------------------------------+---------------------------+
{'$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 - 09:01:37:
INFO - 09:01:37: *** Start DOEScenario execution ***
INFO - 09:01:37: DOEScenario
INFO - 09:01:37: Disciplines: func
INFO - 09:01:37: MDO formulation: DisciplinaryOpt
INFO - 09:01:37: Optimization problem:
INFO - 09:01:37: minimize y_1(x_1, x_2)
INFO - 09:01:37: with respect to x_1, x_2
INFO - 09:01:37: over the design space:
INFO - 09:01:37: +------+-------------+-------+-------------+-------+
INFO - 09:01:37: | Name | Lower bound | Value | Upper bound | Type |
INFO - 09:01:37: +------+-------------+-------+-------------+-------+
INFO - 09:01:37: | x_1 | 0 | None | 1 | float |
INFO - 09:01:37: | x_2 | 0 | None | 1 | float |
INFO - 09:01:37: +------+-------------+-------+-------------+-------+
INFO - 09:01:37: Solving optimization problem with algorithm fullfact:
INFO - 09:01:37: 11%|█ | 1/9 [00:00<00:00, 286.11 it/sec, obj=1]
INFO - 09:01:37: 22%|██▏ | 2/9 [00:00<00:00, 464.10 it/sec, obj=2]
INFO - 09:01:37: 33%|███▎ | 3/9 [00:00<00:00, 599.70 it/sec, obj=3]
INFO - 09:01:37: 44%|████▍ | 4/9 [00:00<00:00, 711.05 it/sec, obj=2.5]
INFO - 09:01:37: 56%|█████▌ | 5/9 [00:00<00:00, 801.39 it/sec, obj=3.5]
INFO - 09:01:37: 67%|██████▋ | 6/9 [00:00<00:00, 876.46 it/sec, obj=4.5]
INFO - 09:01:37: 78%|███████▊ | 7/9 [00:00<00:00, 936.77 it/sec, obj=4]
INFO - 09:01:37: 89%|████████▉ | 8/9 [00:00<00:00, 989.84 it/sec, obj=5]
INFO - 09:01:37: 100%|██████████| 9/9 [00:00<00:00, 1031.53 it/sec, obj=6]
INFO - 09:01:37: Optimization result:
INFO - 09:01:37: Optimizer info:
INFO - 09:01:37: Status: None
INFO - 09:01:37: Message: None
INFO - 09:01:37: Number of calls to the objective function by the optimizer: 9
INFO - 09:01:37: Solution:
INFO - 09:01:37: Objective: 1.0
INFO - 09:01:37: Design space:
INFO - 09:01:37: +------+-------------+-------+-------------+-------+
INFO - 09:01:37: | Name | Lower bound | Value | Upper bound | Type |
INFO - 09:01:37: +------+-------------+-------+-------------+-------+
INFO - 09:01:37: | x_1 | 0 | 0 | 1 | float |
INFO - 09:01:37: | x_2 | 0 | 0 | 1 | float |
INFO - 09:01:37: +------+-------------+-------+-------------+-------+
INFO - 09:01:37: *** End DOEScenario execution (time: 0:00:00.022058) ***
Total running time of the script: (0 minutes 0.070 seconds)