Rename Y_ to embedding_
Generalize with average of nearest neighbors