tfma.post_export_metrics.confusion_matrix_at_thresholds

This is the function that the user calls.