estimator_results_to_array

tsml_eval.utils.results_loading.estimator_results_to_array(estimator_results, measure, include_missing=False)

Convert a list of EstimatorResults objects to an array of metrics.

Follows the output format of aeon array results loaders.

Parameters:

estimator_resultslist of EstimatorResults

The results to convert.

measurestr

A valid metric attribute from the relevant EstimatorResults object. Dependent on the task, i.e. for classification, “accuracy”, “auroc”, “balacc”, and regression, “mse”, “mae”, “r2”.

include_missingbool, default=False

Whether to include datasets with missing results in the output. If False, the whole problem is ignored if any estimator is missing results it. If True, NaN is returned instead of a score in missing cases.

Returns:

results: 2D numpy array

Array of scores. Each column is a results for an estimator, each row a dataset. For multiple resamples, these scores are averaged.

dataset_names: list of str

List of dataset names that were retained.

estimator_names: list of str

List of estimator names that were retained.