gemseo / core / mdofunctions

function_from_discipline module

The MDOFunction subclass to create a function from an MDODiscipline.

class gemseo.core.mdofunctions.function_from_discipline.FunctionFromDiscipline(output_names, mdo_formulation, discipline=None, top_level_disc=True, x_names=None, all_data_names=None, differentiable=True)[source]

Bases: MDOFunction

An MDOFunction object from an MDODiscipline.

Parameters:
  • output_names (Sequence[str]) – The names of the outputs.

  • mdo_formulation (BaseFormulation) – The MDOFormulation object in which the function is located.

  • discipline (MDODiscipline | None) – The discipline computing these outputs. If None, the discipline is detected from the inner disciplines.

  • top_level_disc (bool) –

    If True, search the discipline among the top level ones.

    By default it is set to True.

  • x_names (Sequence[str] | None) – The names of the design variables. If None, use self.get_x_names_of_disc(discipline).

  • all_data_names (Iterable[str] | None) – The reference data names for masking x. If None, use self.get_optim_variables_names().

  • differentiable (bool) –

    If True, then inputs and outputs are added to the list of variables to be differentiated.

    By default it is set to True.

check_grad(x_vect, method='FirstOrderFD', step=1e-06, error_max=1e-08)

Check the gradients of the function.

Parameters:
  • x_vect (ndarray[Any, dtype[Number]]) – The vector at which the function is checked.

  • method (str) –

    The method used to approximate the gradients, either “FirstOrderFD” or “ComplexStep”.

    By default it is set to “FirstOrderFD”.

  • step (float) –

    The step for the approximation of the gradients.

    By default it is set to 1e-06.

  • error_max (float) –

    The maximum value of the error.

    By default it is set to 1e-08.

Raises:

ValueError – Either if the approximation method is unknown, if the shapes of the analytical and approximated Jacobian matrices are inconsistent or if the analytical gradients are wrong.

Return type:

None

static concatenate(functions, name, f_type=None)

Concatenate functions.

Parameters:
  • functions (Iterable[MDOFunction]) – The functions to be concatenated.

  • name (str) – The name of the concatenation function.

  • f_type (str | None) – The type of the concatenation function. If None, the function will have no type.

Returns:

The concatenation of the functions.

Return type:

Concatenate

convex_linear_approx(x_vect, approx_indexes=None, sign_threshold=1e-09)

Compute a convex linearization of the function.

\(\newcommand{\xref}{\hat{x}}\newcommand{\dim}{d}\) The convex linearization of a function \(f\) at a point \(\xref\) is defined as

\[\begin{split}\newcommand{\partialder}{\frac{\partial f}{\partial x_i}(\xref)} f(x) \approx f(\xref) + \sum_{\substack{i = 1 \\ \partialder > 0}}^{\dim} \partialder \, (x_i - \xref_i) - \sum_{\substack{i = 1 \\ \partialder < 0}}^{\dim} \partialder \, \xref_i^2 \, \left(\frac{1}{x_i} - \frac{1}{\xref_i}\right).\end{split}\]

\(\newcommand{\approxinds}{I}\) Optionally, one may require the convex linearization of \(f\) with respect to a subset of its variables \(x_{i \in \approxinds}\), \(I \subset \{1, \dots, \dim\}\), rather than all of them:

\[\begin{split}f(x) = f(x_{i \in \approxinds}, x_{i \not\in \approxinds}) \approx f(\xref_{i \in \approxinds}, x_{i \not\in \approxinds}) + \sum_{\substack{i \in \approxinds \\ \partialder > 0}} \partialder \, (x_i - \xref_i) - \sum_{\substack{i \in \approxinds \\ \partialder < 0}} \partialder \, \xref_i^2 \, \left(\frac{1}{x_i} - \frac{1}{\xref_i}\right).\end{split}\]
Parameters:
  • x_vect (ArrayType) – The input vector at which to build the convex linearization.

  • approx_indexes (ndarray[bool] | None) – A boolean mask specifying w.r.t. which inputs the function should be approximated. If None, consider all the inputs.

  • sign_threshold (float) –

    The threshold for the sign of the derivatives.

    By default it is set to 1e-09.

Returns:

The convex linearization of the function at the given input vector.

Return type:

ConvexLinearApprox

static deserialize(file_path)

Deserialize a function from a file.

Parameters:

file_path (str | Path) – The path to the file containing the function.

Returns:

The function instance.

Return type:

MDOFunction

evaluate(x_vect)

Evaluate the function and store the dimension of the output space.

Parameters:

x_vect (ndarray[Any, dtype[Number]]) – The value of the inputs of the function.

Returns:

The value of the output of the function.

Return type:

Union[ndarray[Any, dtype[Number]], Number]

static filt_0(arr, floor_value=1e-06)

Set the non-significant components of a vector to zero.

The component of a vector is non-significant if its absolute value is lower than a threshold.

Parameters:
  • arr (ndarray[Any, dtype[Number]]) – The original vector.

  • floor_value (float) –

    The threshold.

    By default it is set to 1e-06.

Returns:

The original vector whose non-significant components have been set at zero.

Return type:

ndarray[Any, dtype[Number]]

classmethod generate_args(input_dim, args=None)

Generate the names of the inputs of the function.

Parameters:
  • input_dim (int) – The dimension of the input space of the function.

  • args (Sequence[str] | None) – The initial names of the inputs of the function. If there is only one name, e.g. ["var"], use this name as a base name and generate the names of the inputs, e.g. ["var!0", "var!1", "var!2"] if the dimension of the input space is equal to 3. If None, use "x" as a base name and generate the names of the inputs, i.e. ["x!0", "x!1", "x!2"].

Returns:

The names of the inputs of the function.

Return type:

Sequence[str]

get_indexed_name(index)

Return the name of function component.

Parameters:

index (int) – The index of the function component.

Returns:

The name of the function component.

Return type:

str

has_args()

Check if the inputs of the function have names.

Returns:

Whether the inputs of the function have names.

Return type:

bool

has_dim()

Check if the dimension of the output space of the function is defined.

Returns:

Whether the dimension of the output space of the function is defined.

Return type:

bool

has_expr()

Check if the function has an expression.

Returns:

Whether the function has an expression.

Return type:

bool

has_f_type()

Check if the function has a type.

Returns:

Whether the function has a type.

Return type:

bool

has_jac()

Check if the function has an implemented Jacobian function.

Returns:

Whether the function has an implemented Jacobian function.

Return type:

bool

has_outvars()

Check if the outputs of the function have names.

Returns:

Whether the outputs of the function have names.

Return type:

bool

static init_from_dict_repr(**attributes)

Initialize a new function.

This is typically used for deserialization.

Parameters:

**attributes – The values of the serializable attributes listed in MDOFunction.DICT_REPR_ATTR.

Returns:

A function initialized from the provided data.

Raises:

ValueError – If the name of an argument is not in MDOFunction.DICT_REPR_ATTR.

Return type:

MDOFunction

is_constraint()

Check if the function is a constraint.

The type of a constraint function is either ‘eq’ or ‘ineq’.

Returns:

Whether the function is a constraint.

Return type:

bool

linear_approximation(x_vect, name=None, f_type=None, args=None)

Compute a first-order Taylor polynomial of the function.

\(\newcommand{\xref}{\hat{x}}\newcommand{\dim}{d}\) The first-order Taylor polynomial of a (possibly vector-valued) function \(f\) at a point \(\xref\) is defined as

\[\newcommand{\partialder}{\frac{\partial f}{\partial x_i}(\xref)} f(x) \approx f(\xref) + \sum_{i = 1}^{\dim} \partialder \, (x_i - \xref_i).\]
Parameters:
  • x_vect (ArrayType) – The input vector at which to build the Taylor polynomial.

  • name (str | None) – The name of the linear approximation function. If None, create a name from the name of the function.

  • f_type (str | None) – The type of the linear approximation function. If None, the function will have no type.

  • args (Sequence[str] | None) – The names of the inputs of the linear approximation function, or a name base. If None, use the names of the inputs of the function.

Returns:

The first-order Taylor polynomial of the function at the input vector.

Return type:

MDOLinearFunction

offset(value)

Add an offset value to the function.

Parameters:

value (Union[ndarray[Any, dtype[Number]], Number]) – The offset value.

Returns:

The offset function.

Return type:

MDOFunction

quadratic_approx(x_vect, hessian_approx, args=None)

Build a quadratic approximation of the function at a given point.

The function must be scalar-valued.

\(\newcommand{\xref}{\hat{x}}\newcommand{\dim}{d}\newcommand{ \hessapprox}{\hat{H}}\) For a given approximation \(\hessapprox\) of the Hessian matrix of a function \(f\) at a point \(\xref\), the quadratic approximation of \(f\) is defined as

\[\newcommand{\partialder}{\frac{\partial f}{\partial x_i}(\xref)} f(x) \approx f(\xref) + \sum_{i = 1}^{\dim} \partialder \, (x_i - \xref_i) + \frac{1}{2} \sum_{i = 1}^{\dim} \sum_{j = 1}^{\dim} \hessapprox_{ij} (x_i - \xref_i) (x_j - \xref_j).\]
Parameters:
  • x_vect (ArrayType) – The input vector at which to build the quadratic approximation.

  • hessian_approx (ArrayType) – The approximation of the Hessian matrix at this input vector.

  • args (Sequence[str] | None) – The names of the inputs of the quadratic approximation function, or a name base. If None, use the ones of the current function.

Returns:

The second-order Taylor polynomial of the function at the given point.

Raises:
  • ValueError – Either if the approximated Hessian matrix is not square, or if it is not consistent with the dimension of the given point.

  • AttributeError – If the function does not have an implemented Jacobian function.

Return type:

MDOQuadraticFunction

static rel_err(a_vect, b_vect, error_max)

Compute the 2-norm of the difference between two vectors.

Normalize it with the 2-norm of the reference vector if the latter is greater than the maximal error.

Parameters:
Returns:

The difference between two vectors, normalized if required.

Return type:

float

restrict(frozen_indexes, frozen_values, input_dim, name=None, f_type=None, expr=None, args=None)

Return a restriction of the function

\(\newcommand{\frozeninds}{I}\newcommand{\xfrozen}{\hat{x}}\newcommand{ \frestr}{\hat{f}}\) For a subset \(\approxinds\) of the variables indexes of a function \(f\) to remain frozen at values \(\xfrozen_{i \in \frozeninds}\) the restriction of \(f\) is given by

\[\frestr: x_{i \not\in \approxinds} \longmapsto f(\xref_{i \in \approxinds}, x_{i \not\in \approxinds}).\]
Parameters:
  • frozen_indexes (ndarray[int]) – The indexes of the inputs that will be frozen

  • frozen_values (ArrayType) – The values of the inputs that will be frozen.

  • input_dim (int) – The dimension of input space of the function before restriction.

  • name (str | None) – The name of the function after restriction. If None, create a default name based on the name of the current function and on the argument args.

  • f_type (str | None) – The type of the function after restriction. If None, the function will have no type.

  • expr (str | None) – The expression of the function after restriction. If None, the function will have no expression.

  • args (Sequence[str] | None) – The names of the inputs of the function after restriction. If None, the inputs of the function will have no names.

Returns:

The restriction of the function.

Return type:

FunctionRestriction

serialize(file_path)

Serialize the function and store it in a file.

Parameters:

file_path (str | Path) – The path to the file to store the function.

Return type:

None

set_pt_from_database(database, design_space, normalize=False, jac=True, x_tolerance=1e-10)

Set the original function and Jacobian function from a database.

For a given input vector, the method MDOFunction.func() will return either the output vector stored in the database if the input vector is present or None. The same for the method MDOFunction.jac().

Parameters:
  • database (Database) – The database to read.

  • design_space (DesignSpace) – The design space used for normalization.

  • normalize (bool) –

    If True, the values of the inputs are unnormalized before call.

    By default it is set to False.

  • jac (bool) –

    If True, a Jacobian pointer is also generated.

    By default it is set to True.

  • x_tolerance (float) –

    The tolerance on the distance between inputs.

    By default it is set to 1e-10.

Return type:

None

to_dict()

Create a dictionary representation of the function.

This is used for serialization. The pointers to the functions are removed.

Returns:

Some attributes of the function indexed by their names. See MDOFunction.DICT_REPR_ATTR.

Return type:

dict[str, str | int | list[str]]

AVAILABLE_TYPES: list[str] = ['obj', 'eq', 'ineq', 'obs']

The available types of function.

COEFF_FORMAT_1D: str = '{:.2e}'

The format to be applied to a number when represented in a vector.

COEFF_FORMAT_ND: str = '{: .2e}'

The format to be applied to a number when represented in a matrix.

DEFAULT_ARGS_BASE: str = 'x'

The default name base for the inputs.

DICT_REPR_ATTR: list[str] = ['name', 'f_type', 'expr', 'args', 'dim', 'special_repr']

The names of the attributes to be serialized.

INDEX_PREFIX: str = '!'

The character used to separate a name base and a prefix, e.g. "x!1.

TYPE_EQ: str = 'eq'

The type of function for equality constraint.

TYPE_INEQ: str = 'ineq'

The type of function for inequality constraint.

TYPE_OBJ: str = 'obj'

The type of function for objective.

TYPE_OBS: str = 'obs'

The type of function for observable.

activate_counters: ClassVar[bool] = True

Whether to count the number of function evaluations.

property args: list[str]

The names of the inputs of the function.

Use a copy of the original names.

property default_repr: str

The default string representation of the function.

property dim: int

The dimension of the output space of the function.

property expects_normalized_inputs: bool

Whether the functions expect normalized inputs or not.

property expr: str

The expression of the function, e.g. “2*x”.

property f_type: str

The type of the function, among MDOFunction.AVAILABLE_TYPES.

force_real: bool

Whether to cast the results to real value.

property func: Callable[[ndarray[Any, dtype[Number]]], Union[ndarray[Any, dtype[Number]], Number]]

The function to be evaluated from a given input vector.

has_default_name: bool

Whether the name has been set with a default value.

property jac: Callable[[ndarray[Any, dtype[Number]]], ndarray[Any, dtype[Number]]]

The Jacobian function to be evaluated from a given input vector.

last_eval: OutputType | None

The value of the function output at the last evaluation.

None if it has not yet been evaluated.

property n_calls: int

The number of times the function has been evaluated.

This count is both multiprocess- and multithread-safe, thanks to the locking process used by MDOFunction.evaluate().

property name: str

The name of the function.

property outvars: list[str]

The names of the outputs of the function.

Use a copy of the original names.

special_repr: str

The string representation of the function overloading its default string ones.