ertk.train.get_scores

ertk.train.get_scores(scoring: str | list[str] | dict[str, collections.abc.Callable[[numpy.ndarray, numpy.ndarray], float]] | Callable[[...], float], y_pred: ndarray, y_true: ndarray) → dict[str, Any]

Get dictionary of scores for predictions.

Parameters:

scoring: str, list, dict or callable

Score(s) to calculate. This takes the same for as for scikit-learn’s cross_val_* methods.

y_pred: array-like

Predictions.

y_true: array-like

Ground truth.

Returns:

scores: dict

A dictionary mapping score names to corresponding score(s).