hookeai.utilities.prediction_metrics.compute_prediction_metrics

compute_prediction_metrics(sample_prediction_path, prediction_metrics=[])[source]

Compute prediction metrics from sample prediction file path.

Parameters:
  • sample_prediction_path (str) – Sample prediction file path.

  • prediction_metrics (list[str]) – Prediction metrics.

Returns:

prediction_metrics_results – Prediction metrics results (item, torch.Tensor) for each prediction metric (key, str).

Return type:

dict