tfma.post_export_metrics.mean_squared_error

This is the function that the user calls.