I think this makes more sense as a function of `X` and `X_embedded`.