kNN baselines with original embeddings for XYw and XYb DONE