weights for compute_metrics