gemseo.mlearning.regression.quality.rmse_measure module#

The root mean squared error to assess the quality of a regressor.

The root mean squared error (RMSE) is defined by

\[\begin{split}\\operatorname{RMSE}(\\hat{y})=\\sqrt{\\frac{1}{n}\\sum_{i=1}^n(\\hat{y}_i-y_i)^2},\end{split}\]

where \(\\hat{y}\) are the predictions and \(y\) are the data points.

class RMSEMeasure(algo, fit_transformers=True)[source]#

Bases: MSEMeasure

The root mean squared error to assess the quality of a regressor.

Parameters:

  • algo (BaseRegressor) -- A machine learning algorithm for regression.

  • fit_transformers (bool) --

    Whether to re-fit the transformers when using resampling techniques. If False, use the transformers of the algorithm fitted from the whole learning dataset.

    By default it is set to True.

compute_bootstrap_measure(n_replicates=100, samples=(), multioutput=True, seed=None, as_dict=False, store_resampling_result=False)[source]#

Evaluate the quality of the ML model using the bootstrap technique.

Parameters:

  • n_replicates (int) --

    The number of bootstrap replicates.

    By default it is set to 100.

  • samples (Sequence[int]) --

    The indices of the learning samples. If empty, use the whole learning dataset.

    By default it is set to ().

  • multioutput (bool) --

    Whether the quality measure is returned for each component of the outputs. Otherwise, the average quality measure.

    By default it is set to True.

  • seed (int | None) -- The seed of the pseudo-random number generator. If None, an unpredictable generator will be used.

  • as_dict (bool) --

    Whether the full quality measure is returned as a mapping from algo.output names to quality measures. Otherwise, the full quality measure as an array stacking these quality measures according to the order of algo.output_names.

    By default it is set to False.

  • store_resampling_result (bool) --

    Whether to store the \(n\) machine learning algorithms and associated predictions generated by the resampling stage where \(n\) is the number of bootstrap replicates.

    By default it is set to False.

Returns:

The quality of the ML model.

Return type:

MeasureType

compute_cross_validation_measure(n_folds=5, samples=(), multioutput=True, randomize=True, seed=None, as_dict=False, store_resampling_result=False)[source]#

Evaluate the quality of the ML model using the k-folds technique.

Parameters:

  • n_folds (int) --

    The number of folds.

    By default it is set to 5.

  • samples (Sequence[int]) --

    The indices of the learning samples. If empty, use the whole learning dataset.

    By default it is set to ().

  • multioutput (bool) --

    Whether the quality measure is returned for each component of the outputs. Otherwise, the average quality measure.

    By default it is set to True.

  • randomize (bool) --

    Whether to shuffle the samples before dividing them in folds.

    By default it is set to True.

  • seed (int | None) -- The seed of the pseudo-random number generator. If None, an unpredictable generator is used.

  • as_dict (bool) --

    Whether the full quality measure is returned as a mapping from algo.output names to quality measures. Otherwise, the full quality measure as an array stacking these quality measures according to the order of algo.output_names.

    By default it is set to False.

  • store_resampling_result (bool) --

    Whether to store the \(n\) machine learning algorithms and associated predictions generated by the resampling stage where \(n\) is the number of folds.

    By default it is set to False.

Returns:

The quality of the ML model.

Return type:

MeasureType

compute_learning_measure(samples=(), multioutput=True, as_dict=False)[source]#

Evaluate the quality of the ML model from the learning dataset.

Parameters:

  • samples (Sequence[int]) --

    The indices of the learning samples. If empty, use the whole learning dataset.

    By default it is set to ().

  • multioutput (bool) --

    Whether the quality measure is returned for each component of the outputs. Otherwise, the average quality measure.

    By default it is set to True.

  • as_dict (bool) --

    Whether the full quality measure is returned as a mapping from algo.output names to quality measures. Otherwise, the full quality measure as an array stacking these quality measures according to the order of algo.output_names.

    By default it is set to False.

Returns:

The quality of the ML model.

Return type:

MeasureType

compute_test_measure(test_data, samples=(), multioutput=True, as_dict=False)[source]#

Evaluate the quality of the ML model from a test dataset.

Parameters:

  • test_data (IODataset) -- The test dataset.

  • samples (Sequence[int]) --

    The indices of the learning samples. If empty, use the whole learning dataset.

    By default it is set to ().

  • multioutput (bool) --

    Whether the quality measure is returned for each component of the outputs. Otherwise, the average quality measure.

    By default it is set to True.

  • as_dict (bool) --

    Whether the full quality measure is returned as a mapping from algo.output names to quality measures. Otherwise, the full quality measure as an array stacking these quality measures according to the order of algo.output_names.

    By default it is set to False.

Returns:

The quality of the ML model.

Return type:

MeasureType