tabensemb.trainer.Trainer._metrics#

method

Trainer._metrics(predictions: Dict[str, Dict[str, Tuple[ndarray, ndarray]]], metrics: List[str], test_data_only: bool) → DataFrame[source]#

Calculate metrics for predictions from tabensemb.model.AbstractModel._predict_all().

Parameters:

predictions: Results from tabensemb.model.AbstractModel._predict_all().
metrics: The metrics that have been implemented in tabensemb.utils.utils.metric_sklearn().
test_data_only: Whether to evaluate models only on testing datasets.

Returns:

pd.DataFrame: A dataframe of metrics.