ertk.train.get_scores

ertk.train.get_scores(scoring: str | List[str] | Dict[str, Callable[[ndarray, 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).