ncem.train.GridSearchContainer.get_best_model_id

GridSearchContainer.get_best_model_id(subset_hyperparameters=None, metric_select='r_squared_linreg', partition_select='test', cv_mode='mean')

Get best model identifier.

Parameters

• subset_hyperparameters (list, optional) – List of subset hyperparameters.

• metric_select (str) – Selected metric.

• partition_select (str) – Selected parition.

• cv_mode (str) – cross validation mode.

Return type

best_model_id

Raises

• ValueError – If measure, partition or cv_mode not recognized.

• Warning – If cv_mode max is selected with the following metrics: loss, elbo, mse, mae