from __future__ import annotations

from gemseo.mlearning.api import create_regression_model
from gemseo.mlearning.qual_measure.f1_measure import F1Measure
from gemseo.mlearning.qual_measure.mse_measure import MSEMeasure
from gemseo.mlearning.qual_measure.silhouette import SilhouetteMeasure


In this example, we seek to estimate the Rosenbrock function from the RosenbrockDataset.

dataset = load_dataset("RosenbrockDataset", opt_naming=False)


For that purpose, we will use a MOERegressor in an advanced way: we will not set the clustering, classification and regression algorithms but select them according to their performance from several candidates that we will provide. Moreover, for a given candidate, we will propose several settings, compare their performances and select the best one.

## Initialization¶

First, we initialize a MOERegressor with soft classification by means of the machine learning API function create_regression_model().

model = create_regression_model("MOERegressor", dataset, hard=False)


## Clustering¶

Then, we add two clustering algorithms with different numbers of clusters (called components for the Gaussian Mixture) and set the SilhouetteMeasure as clustering measure to be evaluated from the learning set. During the learning stage, the mixture of experts will select the clustering algorithm and the number of clusters minimizing this measure.

model.set_clustering_measure(SilhouetteMeasure)


## Classification¶

We also add classification algorithms with different settings and set the F1Measure as classification measure to be evaluated from the learning set. During the learning stage, the mixture of experts will select the classification algorithm and the settings minimizing this measure.

model.set_classification_measure(F1Measure)


## Regression¶

We also add regression algorithms and set the MSEMeasure as regression measure to be evaluated from the learning set. During the learning stage, for each cluster, the mixture of experts will select the regression algorithm minimizing this measure.

model.set_regression_measure(MSEMeasure)


Note

We could also add candidates for some learning stages, e.g. clustering and regression, and set the machine learning algorithms for the remaining ones, e.g. classification.

## Training¶

Lastly, we learn the data and select the best machine learning algorithm for both clustering, classification and regression steps.

model.learn()


## Result¶

We can get information on this model, on the sub-machine learning models selected among the candidates and on their selected settings. We can see that a KMeans with four clusters has been selected for the clustering stage, as well as a RandomForestClassifier for the classification stage and a RBFRegressor for each cluster.

print(model)

MOERegressor(hard=False)
built from 100 learning samples
Clustering
KMeans(n_clusters=4, random_state=0, var_names=None)
Classification
RandomForestClassifier(n_estimators=100)
Regression
Local model 0
Local model 1
Local model 2
Local model 3


Note

By adding candidates, and depending on the complexity of the function to be approximated, one could obtain different regression models according to the clusters. For example, one could use a PolynomialRegressor with order 2 on a sub-part of the input space and a GaussianProcessRegressor on another sub-part of the input space.

Once built, this mixture of experts can be used as any MLRegressionAlgo.

Another example proposes a standard use of MOERegressor.