Well done but be careful, TSNE has some drawbacks, well detailed here: https://distill.pub/2016/misread-tsne/ https://distill.pub/2016/misread-tsne/ For buildi

Well done but be careful, TSNE has some drawbacks, well detailed here: https://distill.pub/2016/misread-tsne/ https://distill.pub/2016/misread-tsne/ For building a more precise 2D embedding or image similarity search engine you may want to rely on a neural embedding + FAISS or Annoy instead of compressing into 2D with TSNE. When doing the later, distances don't mean much anymore.